Add one unit test to cover empty tuple

apache · Aug 9, 2019 · 06e264a · 06e264a
1 parent 889e99d
commit 06e264a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py
@@ -206,13 +206,13 @@ def test_forward_split():
 # ---------
 
 
-def _test_forward_transpose(ishape, axes=None):
+def _test_forward_transpose(ishape, axes=()):
     data = np.random.uniform(size=ishape).astype(np.float32)
 
     with tf.Graph().as_default():
         in_data = array_ops.placeholder(shape=data.shape, dtype=data.dtype)
 
-        if axes is None:
+        if not axes:
             out = array_ops.transpose(in_data)
         else:
             out = array_ops.transpose(in_data, axes)
@@ -227,6 +227,7 @@ def test_forward_transpose():
     _test_forward_transpose((2, 3, 4), (1, 2, 0))
     _test_forward_transpose((2, 3, 4), (0, 1, 2))
     _test_forward_transpose((2, 3, 4, 5), (3, 0, 1, 2))
+    _test_forward_transpose((2, 3, 4, 5), ())
 
 
 #######################################################################