Add some debug information to an assert in example_proto_coder_test.

PiperOrigin-RevId: 424634421
tensorflow · Jan 27, 2022 · 5d7323c · 5d7323c
1 parent 4b824e9
commit 5d7323c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tensorflow_transform/coders/example_proto_coder_test.py b/tensorflow_transform/coders/example_proto_coder_test.py
@@ -349,7 +349,10 @@ class ExampleProtoCoderTest(test_case.TransformTestCase):
   def setUp(self):
     super().setUp()
     # Verify that the implementation we requested via the Flag is honoured.
-    assert api_implementation.Type() == flags.FLAGS.proto_implementation_type
+    assert api_implementation.Type() == flags.FLAGS.proto_implementation_type, (
+        'Expected proto implementation type '
+        f'"{flags.FLAGS.proto_implementation_type}", got: '
+        f'"{api_implementation.Type()}"')
 
   def assertSerializedProtosEqual(self, a, b):
     np.testing.assert_equal(_binary_to_example(a), _binary_to_example(b))