mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-02-21 08:42:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'reduce_cyclomatic_complexity_of_split_matrix_function' of github.com:ivanz851/Algorithms-Python into reduce_cyclomatic_complexity_of_split_matrix_function

Browse Source
This commit is contained in:
ivanz-thinkpad 2025-01-31 23:36:58 +03:00
commit 4ceb811be1
1 changed files with 6 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
divide_and_conquer/tests/test_strassen_matrix_multiplication.py
View File

@ -4,18 +4,8 @@ from divide_and_conquer.strassen_matrix_multiplication import split_matrix
def test_4x4_matrix():
matrix = [
[4, 3, 2, 4],
[2, 3, 1, 1],
[6, 5, 4, 3],
[8, 4, 1, 6]
]
expected = (
[[4, 3], [2, 3]],
[[2, 4], [1, 1]],
[[6, 5], [8, 4]],
[[4, 3], [1, 6]]
)
matrix = [[4, 3, 2, 4], [2, 3, 1, 1], [6, 5, 4, 3], [8, 4, 1, 6]]
expected = ([[4, 3], [2, 3]], [[2, 4], [1, 1]], [[6, 5], [8, 4]], [[4, 3], [1, 6]])
assert split_matrix(matrix) == expected
@ -28,23 +18,19 @@ def test_8x8_matrix():
[4, 3, 2, 4, 4, 3, 2, 4],
[2, 3, 1, 1, 2, 3, 1, 1],
[6, 5, 4, 3, 6, 5, 4, 3],
[8, 4, 1, 6, 8, 4, 1, 6]
[8, 4, 1, 6, 8, 4, 1, 6],
]
expected = (
[[4, 3, 2, 4], [2, 3, 1, 1], [6, 5, 4, 3], [8, 4, 1, 6]],
[[4, 3, 2, 4], [2, 3, 1, 1], [6, 5, 4, 3], [8, 4, 1, 6]],
[[4, 3, 2, 4], [2, 3, 1, 1], [6, 5, 4, 3], [8, 4, 1, 6]],
[[4, 3, 2, 4], [2, 3, 1, 1], [6, 5, 4, 3], [8, 4, 1, 6]]
[[4, 3, 2, 4], [2, 3, 1, 1], [6, 5, 4, 3], [8, 4, 1, 6]],
)
assert split_matrix(matrix) == expected
def test_invalid_odd_matrix():
matrix = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
with pytest.raises(Exception, match="Odd matrices are not supported!"):
split_matrix(matrix)
@ -55,7 +41,7 @@ def test_invalid_non_square_matrix():
[5, 6, 7, 8],
[9, 10, 11, 12],
[13, 14, 15, 16],
[17, 18, 19, 20]
[17, 18, 19, 20],
]
with pytest.raises(Exception, match="Odd matrices are not supported!"):
split_matrix(matrix)