c4f3799d5713d467ca577a6d0593f242076c9540,tensorflow_datasets/core/test_utils.py,FeatureExpectationsTestCase,test_encode_decode,#FeatureExpectationsTestCase#,65

Before Change

  @tf.contrib.eager.run_test_in_graph_and_eager_modes()
  def test_encode_decode(self):
    expectations = self.expectations
    fdict = features.FeaturesDict(
        {exp.name: exp.feature for exp in expectations})

    decoded_sample = features_encode_decode(
        fdict, dict([(exp.name, exp.value) for exp in expectations]))

    for exp in expectations:
      self.assertAllEqual(decoded_sample[exp.name], exp.expected)
      // TODO(rsepassi): test shape and dtype against exp.feature

After Change

    // Maybe should try to use metaclass instead and dynamically generate one
    // method per feature expectation.
    for exp in self.expectations:
      tf.logging.info("Testing feature %s", exp.name)

      // Check the shape/dtype
      self.assertEqual(exp.feature.shape, exp.shape)
      self.assertEqual(exp.feature.dtype, exp.dtype)

Instances