from tensorflow.python.platform import test

      self.assertListEqual([height, width, 3], list(image.shape))
      self.assertListEqual([1], list(label.shape))

  def testConflictingRecordKeyItem(self):
    dataset_dir = tempfile.mkdtemp(prefix=os.path.join(self.get_temp_dir(),
                                                       'tfrecord_dataset'))

    with self.cached_session():
      with self.assertRaises(ValueError):
        dataset_data_provider.DatasetDataProvider(
            _create_tfrecord_dataset(dataset_dir), record_key='image')

if __name__ == '__main__':
  test.main()
