diff options
author | 2016-03-08 12:28:54 -0800 | |
---|---|---|
committer | 2016-03-08 17:19:45 -0800 | |
commit | d6192acccda3c6f2c9cbc3456856e764a82aa100 (patch) | |
tree | cd06a8a7af073090675063be5b2556d918b96217 /tensorflow/python/kernel_tests/decode_csv_op_test.py | |
parent | 93766a718b5309343e011c9efbc50096ddb3b67a (diff) |
Add bounds check in DecodeCsvOp for when a quoted string is not closed.
Change: 116684950
Diffstat (limited to 'tensorflow/python/kernel_tests/decode_csv_op_test.py')
-rw-r--r-- | tensorflow/python/kernel_tests/decode_csv_op_test.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/python/kernel_tests/decode_csv_op_test.py b/tensorflow/python/kernel_tests/decode_csv_op_test.py index 307c84f294..959268a544 100644 --- a/tensorflow/python/kernel_tests/decode_csv_op_test.py +++ b/tensorflow/python/kernel_tests/decode_csv_op_test.py @@ -169,6 +169,12 @@ class DecodeCSVOpTest(tf.test.TestCase): self._test(args, expected_err_re="Expect 1 fields but have 2 in record 0") + def testShortQuotedString(self): + args = {"records": ["\""], "record_defaults": [["default"]],} + + self._test(args, + expected_err_re="Quoted field has to end with quote followed.*") + if __name__ == "__main__": tf.test.main() |