Add test coverage for example parsing with a feature that is an empty…

… string. PiperOrigin-RevId: 179875758
STCYC · Dec 22, 2017 · 6b8b9e4 · 6b8b9e4
1 parent e38cf4c
commit 6b8b9e4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tensorflow/python/kernel_tests/parse_single_example_op_test.py b/tensorflow/python/kernel_tests/parse_single_example_op_test.py
@@ -410,7 +410,7 @@ def testSerializedContainingDense(self):
             bname: bytes_feature([b"b0_str"]),
         })), example(features=features({
             aname: float_feature([-1, -1]),
-            bname: bytes_feature([b"b1"]),
+            bname: bytes_feature([b""]),
         }))
     ]
 
@@ -419,7 +419,7 @@ def testSerializedContainingDense(self):
         bname: np.array(["b0_str"], dtype=bytes).reshape(1, 1, 1, 1)
     }, {
         aname: np.array([-1, -1], dtype=np.float32).reshape(1, 2, 1),
-        bname: np.array(["b1"], dtype=bytes).reshape(1, 1, 1, 1)
+        bname: np.array([""], dtype=bytes).reshape(1, 1, 1, 1)
     }]
 
     for proto, expected_output in zip(original, expected_outputs):

diff --git a/tensorflow/python/kernel_tests/parsing_ops_test.py b/tensorflow/python/kernel_tests/parsing_ops_test.py
@@ -426,7 +426,7 @@ def testSerializedContainingDense(self):
             bname: bytes_feature([b"b0_str"]),
         })), example(features=features({
             aname: float_feature([-1, -1]),
-            bname: bytes_feature([b"b1"]),
+            bname: bytes_feature([b""]),
         }))
     ]
 
@@ -438,7 +438,7 @@ def testSerializedContainingDense(self):
                 [[1, 1], [-1, -1]], dtype=np.float32).reshape(2, 1, 2, 1),
         bname:
             np.array(
-                ["b0_str", "b1"], dtype=bytes).reshape(2, 1, 1, 1, 1),
+                ["b0_str", ""], dtype=bytes).reshape(2, 1, 1, 1, 1),
     }
 
     # No defaults, values required