Added test to check if shape returns tuple

YangWeipengscut · Jan 8, 2022 · f9baf2d · f9baf2d
1 parent 87b435b
commit f9baf2d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorly/tests/test_backend.py b/tensorly/tests/test_backend.py
@@ -216,6 +216,8 @@ def test_shape():
     A2 = T.reshape(A, shape2)
     assert_equal(T.shape(A2), shape2)
 
+    assert type(T.shape(A2)) == tuple
+
 
 def test_ndim():
     A = T.arange(3*4*5)