File tree Expand file tree Collapse file tree 1 file changed +15
-0
lines changed Expand file tree Collapse file tree 1 file changed +15
-0
lines changed Original file line number Diff line number Diff line change @@ -14,6 +14,7 @@ def check_mat_and_vec(mat: list[list[f64]], vec: list[f64]):
14
14
assert vec [i ] == 2 * float (i )
15
15
16
16
def test_list_of_lists ():
17
+ tensors : list [list [list [list [f64 ]]]] = []
17
18
tensor : list [list [list [f64 ]]] = []
18
19
mat : list [list [f64 ]] = []
19
20
vec : list [f64 ] = []
@@ -22,6 +23,7 @@ def test_list_of_lists():
22
23
i : i32
23
24
j : i32
24
25
k : i32
26
+ l : i32
25
27
26
28
for i in range (rows ):
27
29
for j in range (cols ):
@@ -45,4 +47,17 @@ def test_list_of_lists():
45
47
for j in range (cols ):
46
48
assert mat [i ][j ] - tensor [k ][i ][j ] == rows - k
47
49
50
+ for l in range (2 * rows ):
51
+ tensors .append (tensor )
52
+ for i in range (rows ):
53
+ for j in range (rows ):
54
+ for k in range (cols ):
55
+ tensor [i ][j ][k ] += float (1 )
56
+
57
+ for l in range (2 * rows ):
58
+ for i in range (rows ):
59
+ for j in range (rows ):
60
+ for k in range (cols ):
61
+ assert tensor [i ][j ][k ] - tensors [l ][i ][j ][k ] == 2 * rows - l
62
+
48
63
test_list_of_lists ()
You can’t perform that action at this time.
0 commit comments