3
3
4
4
5
5
def test_4x4_matrix ():
6
- matrix = [
7
- [4 , 3 , 2 , 4 ],
8
- [2 , 3 , 1 , 1 ],
9
- [6 , 5 , 4 , 3 ],
10
- [8 , 4 , 1 , 6 ]
11
- ]
12
- expected = (
13
- [[4 , 3 ], [2 , 3 ]],
14
- [[2 , 4 ], [1 , 1 ]],
15
- [[6 , 5 ], [8 , 4 ]],
16
- [[4 , 3 ], [1 , 6 ]]
17
- )
6
+ matrix = [[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]]
7
+ expected = ([[4 , 3 ], [2 , 3 ]], [[2 , 4 ], [1 , 1 ]], [[6 , 5 ], [8 , 4 ]], [[4 , 3 ], [1 , 6 ]])
18
8
assert split_matrix (matrix ) == expected
19
9
20
10
@@ -27,23 +17,19 @@ def test_8x8_matrix():
27
17
[4 , 3 , 2 , 4 , 4 , 3 , 2 , 4 ],
28
18
[2 , 3 , 1 , 1 , 2 , 3 , 1 , 1 ],
29
19
[6 , 5 , 4 , 3 , 6 , 5 , 4 , 3 ],
30
- [8 , 4 , 1 , 6 , 8 , 4 , 1 , 6 ]
20
+ [8 , 4 , 1 , 6 , 8 , 4 , 1 , 6 ],
31
21
]
32
22
expected = (
33
23
[[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]],
34
24
[[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]],
35
25
[[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]],
36
- [[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]]
26
+ [[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]],
37
27
)
38
28
assert split_matrix (matrix ) == expected
39
29
40
30
41
31
def test_invalid_odd_matrix ():
42
- matrix = [
43
- [1 , 2 , 3 ],
44
- [4 , 5 , 6 ],
45
- [7 , 8 , 9 ]
46
- ]
32
+ matrix = [[1 , 2 , 3 ], [4 , 5 , 6 ], [7 , 8 , 9 ]]
47
33
with pytest .raises (Exception , match = "Odd matrices are not supported!" ):
48
34
split_matrix (matrix )
49
35
@@ -54,7 +40,7 @@ def test_invalid_non_square_matrix():
54
40
[5 , 6 , 7 , 8 ],
55
41
[9 , 10 , 11 , 12 ],
56
42
[13 , 14 , 15 , 16 ],
57
- [17 , 18 , 19 , 20 ]
43
+ [17 , 18 , 19 , 20 ],
58
44
]
59
45
with pytest .raises (Exception , match = "Odd matrices are not supported!" ):
60
46
split_matrix (matrix )
0 commit comments