experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add test when axis is an empty list

Browse Source
This commit is contained in:
Tzu-Wei Sung 2020-12-08 20:52:29 -08:00
parent f3b09dcc6a
commit a49b21b06e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/kernel_tests/unique_op_test.py
View File

@ -123,6 +123,8 @@ class UniqueTest(test.TestCase):
self.assertEqual(idx.shape.as_list(), [2])
_, idx = gen_array_ops.unique_v2(x, axis=[-3])
self.assertEqual(idx.shape.as_list(), [3])
_, idx = gen_array_ops.unique_v2([0, 1, 2], axis=[])
self.assertEqual(idx.shape.as_list(), [3])
with self.assertRaisesRegexp(ValueError, "axis expects a 1D vector"):
gen_array_ops.unique_v2(x, axis=[[0]])