11from lpython import c64 , i32
22from copy import deepcopy
33
4- def generate_complex_tensors (mat : list [list [c64 ]], vec : list [c64 ]) -> list [tuple [list [list [c64 ]], list [c64 ]]]:
5- tensor : tuple [list [list [c64 ]], list [c64 ]]
6- tensors : list [tuple [list [list [c64 ]], list [c64 ]]] = []
4+ def generate_complex_arrays (mat : list [list [c64 ]], vec : list [c64 ]) -> list [tuple [list [list [c64 ]], list [c64 ]]]:
5+ array : tuple [list [list [c64 ]], list [c64 ]]
6+ arrays : list [tuple [list [list [c64 ]], list [c64 ]]] = []
77 rows : i32 = len (mat )
88 cols : i32 = len (vec )
99 i : i32 ; j : i32 ; k : i32
1010
11- tensor = (deepcopy (mat ), deepcopy (vec ))
11+ array = (deepcopy (mat ), deepcopy (vec ))
1212
1313 for k in range (2 * rows ):
14- tensors .append (deepcopy (tensor ))
14+ arrays .append (deepcopy (array ))
1515 for i in range (rows ):
1616 for j in range (cols ):
1717 mat [i ][j ] += complex (1.0 , 2.0 )
1818
1919 for i in range (cols ):
2020 vec [i ] += complex (1.0 , 2.0 )
2121
22- tensor = (deepcopy (mat ), deepcopy (vec ))
22+ array = (deepcopy (mat ), deepcopy (vec ))
2323
24- return tensors
24+ return arrays
2525
2626def test_tuple_with_lists ():
2727 mat : list [list [c64 ]] = []
2828 vec : list [c64 ] = []
29- tensor : tuple [list [list [c64 ]], list [c64 ]]
30- tensors : list [tuple [list [list [c64 ]], list [c64 ]]] = []
29+ array : tuple [list [list [c64 ]], list [c64 ]]
30+ arrays : list [tuple [list [list [c64 ]], list [c64 ]]] = []
3131 i : i32
3232 j : i32
3333 k : i32
@@ -48,7 +48,7 @@ def test_tuple_with_lists():
4848 for j in range (cols ):
4949 assert mat [i ][j ] - vec [j ] == c64 (i - j )
5050
51- tensor = (deepcopy (mat ), deepcopy (vec ))
51+ array = (deepcopy (mat ), deepcopy (vec ))
5252
5353 for i in range (rows ):
5454 for j in range (cols ):
@@ -59,20 +59,20 @@ def test_tuple_with_lists():
5959
6060 for i in range (rows ):
6161 for j in range (cols ):
62- assert tensor [0 ][i ][j ] - mat [i ][j ] == - complex (0 , 3.0 )
62+ assert array [0 ][i ][j ] - mat [i ][j ] == - complex (0 , 3.0 )
6363
6464 for i in range (cols ):
65- assert tensor [1 ][i ] - vec [i ] == - complex (0 , 2.0 )
65+ assert array [1 ][i ] - vec [i ] == - complex (0 , 2.0 )
6666
67- tensors = generate_complex_tensors (mat , vec )
67+ arrays = generate_complex_arrays (mat , vec )
6868
6969 for k in range (2 * rows ):
7070 for i in range (rows ):
7171 for j in range (cols ):
72- assert tensors [k ][0 ][i ][j ] - mat [i ][j ] == - c64 (2 * rows - k ) * complex (1.0 , 2.0 )
72+ assert arrays [k ][0 ][i ][j ] - mat [i ][j ] == - c64 (2 * rows - k ) * complex (1.0 , 2.0 )
7373
7474 for k in range (2 * rows ):
7575 for i in range (cols ):
76- assert tensors [k ][1 ][i ] - vec [i ] == - c64 (2 * rows - k ) * complex (1.0 , 2.0 )
76+ assert arrays [k ][1 ][i ] - vec [i ] == - c64 (2 * rows - k ) * complex (1.0 , 2.0 )
7777
7878test_tuple_with_lists ()
0 commit comments