Skip to content

Commit 3d93067

Browse files
committed
Add back 'complex64' type to tests
1 parent d28d774 commit 3d93067

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/tensor/test_basic.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1006,7 +1006,8 @@ def check(dtype, N, M_=None, k=0):
10061006
"float32",
10071007
"float64",
10081008
"uint16",
1009-
]: # Handle "complex64" ?
1009+
"complex64",
1010+
]:
10101011
check(dtype, 3)
10111012
# M != N, k = 0
10121013
check(dtype, 3, 5)
@@ -1078,7 +1079,7 @@ def check_u_batch(m):
10781079
assert np.allclose(result, np.triu(m, k))
10791080
assert result.dtype == np.dtype(dtype)
10801081

1081-
for dtype in ["int32", "int64", "float32", "float64", "uint16"]:
1082+
for dtype in ["int32", "int64", "float32", "float64", "uint16", "complex64"]:
10821083
m = random_of_dtype((10, 10), dtype)
10831084
check_l(m, 0)
10841085
check_l(m, 1)

0 commit comments

Comments
 (0)