Skip to content

Commit bc57448

Browse files
committed
ready
1 parent fbfbb0f commit bc57448

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

integration_tests/test_list_06.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ def check_mat_and_vec(mat: list[list[f64]], vec: list[f64]):
1414
assert vec[i] == 2 * float(i)
1515

1616
def test_list_of_lists():
17+
tensors: list[list[list[list[f64]]]] = []
1718
tensor: list[list[list[f64]]] = []
1819
mat: list[list[f64]] = []
1920
vec: list[f64] = []
@@ -22,6 +23,7 @@ def test_list_of_lists():
2223
i: i32
2324
j: i32
2425
k: i32
26+
l: i32
2527

2628
for i in range(rows):
2729
for j in range(cols):
@@ -45,4 +47,17 @@ def test_list_of_lists():
4547
for j in range(cols):
4648
assert mat[i][j] - tensor[k][i][j] == rows - k
4749

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+
4863
test_list_of_lists()

0 commit comments

Comments
 (0)