diff options
author | feiquan <feiquan@wacai.com> | 2018-08-14 00:07:28 +0800 |
---|---|---|
committer | feiquan <feiquan@wacai.com> | 2018-08-14 00:07:28 +0800 |
commit | 349d81c80a5b64ae09a36624571ec24d9e7a8b1d (patch) | |
tree | 191cc3ea9228a5fd4d5cfed970ca9b55a5c980e3 /tensorflow/contrib/autograph | |
parent | 22ebbbc60e5d94d67cdf6c26b44919f7dbb8f600 (diff) |
add test for gen_item_tensor_string
Diffstat (limited to 'tensorflow/contrib/autograph')
-rw-r--r-- | tensorflow/contrib/autograph/operators/slices_test.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/contrib/autograph/operators/slices_test.py b/tensorflow/contrib/autograph/operators/slices_test.py index d4aacb9d20..9c0b2c77a1 100644 --- a/tensorflow/contrib/autograph/operators/slices_test.py +++ b/tensorflow/contrib/autograph/operators/slices_test.py @@ -46,6 +46,13 @@ class SlicesTest(test.TestCase): with self.test_session() as sess: self.assertAllEqual(sess.run(t), [3, 4]) + def test_get_item_tensor_string(self): + initial_str = constant_op.constant("abcd") + t = slices.get_item(initial_str, 1, slices.GetItemOpts(element_dtype=initial_str.dtype)) + + with self.test_session() as sess: + self.assertEqual(sess.run(t), b"b") + if __name__ == '__main__': test.main() |