aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/autograph
diff options
context:
space:
mode:
authorGravatar feiquan <feiquan@wacai.com>2018-08-14 00:07:28 +0800
committerGravatar feiquan <feiquan@wacai.com>2018-08-14 00:07:28 +0800
commit349d81c80a5b64ae09a36624571ec24d9e7a8b1d (patch)
tree191cc3ea9228a5fd4d5cfed970ca9b55a5c980e3 /tensorflow/contrib/autograph
parent22ebbbc60e5d94d67cdf6c26b44919f7dbb8f600 (diff)
add test for gen_item_tensor_string
Diffstat (limited to 'tensorflow/contrib/autograph')
-rw-r--r--tensorflow/contrib/autograph/operators/slices_test.py7
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()