3
3
4
4
5
5
class TestSplitMatrix (unittest .TestCase ):
6
-
7
6
def test_4x4_matrix (self ):
8
- matrix = [
9
- [4 , 3 , 2 , 4 ],
10
- [2 , 3 , 1 , 1 ],
11
- [6 , 5 , 4 , 3 ],
12
- [8 , 4 , 1 , 6 ]
13
- ]
7
+ matrix = [[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]]
14
8
expected = (
15
9
[[4 , 3 ], [2 , 3 ]],
16
10
[[2 , 4 ], [1 , 1 ]],
17
11
[[6 , 5 ], [8 , 4 ]],
18
- [[4 , 3 ], [1 , 6 ]]
12
+ [[4 , 3 ], [1 , 6 ]],
19
13
)
20
14
self .assertEqual (split_matrix (matrix ), expected )
21
15
@@ -28,31 +22,23 @@ def test_8x8_matrix(self):
28
22
[4 , 3 , 2 , 4 , 4 , 3 , 2 , 4 ],
29
23
[2 , 3 , 1 , 1 , 2 , 3 , 1 , 1 ],
30
24
[6 , 5 , 4 , 3 , 6 , 5 , 4 , 3 ],
31
- [8 , 4 , 1 , 6 , 8 , 4 , 1 , 6 ]
25
+ [8 , 4 , 1 , 6 , 8 , 4 , 1 , 6 ],
32
26
]
33
27
expected = (
34
28
[[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]],
35
29
[[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]],
36
30
[[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]],
37
- [[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]]
31
+ [[4 , 3 , 2 , 4 ], [2 , 3 , 1 , 1 ], [6 , 5 , 4 , 3 ], [8 , 4 , 1 , 6 ]],
38
32
)
39
33
self .assertEqual (split_matrix (matrix ), expected )
40
34
41
35
def test_invalid_odd_matrix (self ):
42
- matrix = [
43
- [1 , 2 , 3 ],
44
- [4 , 5 , 6 ],
45
- [7 , 8 , 9 ]
46
- ]
36
+ matrix = [[1 , 2 , 3 ], [4 , 5 , 6 ], [7 , 8 , 9 ]]
47
37
with self .assertRaises (Exception ):
48
38
split_matrix (matrix )
49
39
50
40
def test_invalid_non_square_matrix (self ):
51
- matrix = [
52
- [1 , 2 , 3 , 4 ],
53
- [5 , 6 , 7 , 8 ],
54
- [9 , 10 , 11 , 12 ]
55
- ]
41
+ matrix = [[1 , 2 , 3 , 4 ], [5 , 6 , 7 , 8 ], [9 , 10 , 11 , 12 ]]
56
42
with self .assertRaises (Exception ):
57
43
split_matrix (matrix )
58
44
0 commit comments