08c0e994cc0e20cb576469c01b472a43e3279e86,tensorflow_transform/beam/tft_beam_io/beam_metadata_io_test.py,,,#,33

Before Change

        (None,), dataset_schema.ListColumnRepresentation())
})

_TEST_METADATA = dataset_metadata.DatasetMetadata({
    "fixed_column": dataset_schema.ColumnSchema(
        tf.string, (3,), dataset_schema.FixedColumnRepresentation()),
    // zeros will be overriddden
    "list_columm": dataset_schema.ColumnSchema(
        dataset_schema.IntDomain(tf.int64, min_value=0, max_value=0),
        (None,), dataset_schema.ListColumnRepresentation())
})


class BeamMetadataIoTest(test_util.TensorFlowTestCase):

After Change

    self.assertEqual(metadata, test_metadata.COMPLETE_METADATA)


if __name__ == "__main__":
  unittest.main()

Instances