1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
#!/usr/bin/env python3
import unittest
ch_1 = __import__('ch-1')
ch_2 = __import__('ch-2')
class TestClass(unittest.TestCase):
def test_ch_1(self):
matrix_1 = [
[4, 4, 4, 4],
[10, 0, 0, 0],
[2, 2, 2, 9]
]
matrix_2 = [[1, 5],[7, 3], [3, 5]]
matrix_3 = [[1, 2, 3], [3, 2, 1]]
matrix_4 = [[2, 8, 7], [7, 1, 3], [1, 9, 5]]
matrix_5 = [
[10, 20, 30],
[5, 5, 5],
[0, 100, 0],
[25, 25, 25]
]
self.assertEqual(ch_1.highest_row(matrix_1), 16)
self.assertEqual(ch_1.highest_row(matrix_2), 10)
self.assertEqual(ch_1.highest_row(matrix_3), 6)
self.assertEqual(ch_1.highest_row(matrix_4), 17)
self.assertEqual(ch_1.highest_row(matrix_5), 100)
def test_ch_2(self):
self.assertEqual(ch_2.max_distance([4, 5, 7],[9, 1, 3, 4]), 6)
self.assertEqual(ch_2.max_distance([2, 3, 5, 4],[3, 2, 5, 5, 8, 7]), 6)
self.assertEqual(ch_2.max_distance([2, 1, 11, 3],[2, 5, 10, 2]), 9)
self.assertEqual(ch_2.max_distance([1, 2, 3],[3, 2, 1]), 2)
self.assertEqual(ch_2.max_distance([1, 0, 2, 3],[5, 0]), 5)
if __name__ == '__main__':
unittest.main()
|
