Skip to content

Commit f5fc9df

Browse files
committed
Modify InferShape test for Tri class
1 parent 3cc8e68 commit f5fc9df

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/tensor/test_basic.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3930,15 +3930,15 @@ def test_Tri(self):
39303930
biscal = iscalar()
39313931
ciscal = iscalar()
39323932
self._compile_and_check(
3933-
[aiscal, biscal, ciscal], [Tri()(aiscal, biscal, ciscal)], [4, 4, 0], Tri
3933+
[aiscal, biscal, ciscal], [tri(aiscal, biscal, ciscal)], [4, 4, 0], Tri
39343934
)
39353935

39363936
self._compile_and_check(
3937-
[aiscal, biscal, ciscal], [Tri()(aiscal, biscal, ciscal)], [4, 5, 0], Tri
3937+
[aiscal, biscal, ciscal], [tri(aiscal, biscal, ciscal)], [4, 5, 0], Tri
39383938
)
39393939

39403940
self._compile_and_check(
3941-
[aiscal, biscal, ciscal], [Tri()(aiscal, biscal, ciscal)], [3, 5, 0], Tri
3941+
[aiscal, biscal, ciscal], [tri(aiscal, biscal, ciscal)], [3, 5, 0], Tri
39423942
)
39433943

39443944
def test_ExtractDiag(self):

0 commit comments

Comments
 (0)