Skip to content

Commit 16124fc

Browse files
committed
fix: failed tests
1 parent 370d5fd commit 16124fc

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

matrix/kronecker_product.py

+5-3
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@ def kronecker_product(
3333
>>> kronecker_product([[1, 2]], [[5, 6], [7, 8]])
3434
[[5, 6, 10, 12], [7, 8, 14, 16]]
3535
36-
>>> kronecker_product([[1, 2, 3], [4, 5, 6]], [[5, 6], [7, 8]])
37-
[[5, 6, 10, 12, 15, 18], [7, 8, 14, 16, 21, 24], [20, 24, 25, 30, 30, 36], [28, 32, 35, 40, 42, 48]]
36+
>>> kronecker_product([[1, 2], [4, 5]], [[5, 6], [7, 8]])
37+
[[5, 6, 10, 12], [7, 8, 14, 16], [20, 24, 25, 30], [28, 32, 35, 40]]
3838
3939
>>> kronecker_product([1, 2], [[5, 6], [7, 8]])
4040
Traceback (most recent call last):
@@ -64,7 +64,9 @@ def kronecker_product(
6464
for c_index_b in range(cols_matrix_b):
6565
result[r_index_a * rows_matrix_b + r_index_b][
6666
c_index_a * cols_matrix_b + c_index_b
67-
] = matrix_a[r_index_a][c_index_a] * matrix_b[r_index_b][c_index_b]
67+
] = (
68+
matrix_a[r_index_a][c_index_a] * matrix_b[r_index_b][c_index_b]
69+
)
6870

6971
return result
7072

0 commit comments

Comments
 (0)