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
|
#!/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):
self.assertEqual(ch_1.maximum_ones([[0, 1],[1, 0]]), 1)
self.assertEqual(ch_1.maximum_ones([[0, 0, 0],[1, 0, 1]]), 2)
self.assertEqual(ch_1.maximum_ones([[0, 0],[1, 1],[0, 0]]), 2)
def test_ch_2(self):
self.assertEqual(
ch_2.sort_by_1_bits([0, 1, 2, 3, 4, 5, 6, 7, 8]),
[0, 1, 2, 4, 8, 3, 5, 6, 7]
)
self.assertEqual(
ch_2.sort_by_1_bits([1024, 512, 256, 128, 64]),
[64, 128, 256, 512, 1024]
)
if __name__ == '__main__':
unittest.main()
|
