1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
#!/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.element_digit_sum([1, 2, 3, 45]), 36)
self.assertEqual(ch_1.element_digit_sum([1, 12, 3]), 9)
self.assertEqual(ch_1.element_digit_sum([1, 2, 3, 4]), 0)
self.assertEqual(ch_1.element_digit_sum([236, 416, 336, 350]), 1296)
def test_ch_2(self):
self.assertEqual(ch_2.multiple_by_two([5, 3, 6, 1, 12], 3), 24)
self.assertEqual(ch_2.multiple_by_two([1, 2, 4, 3], 1), 8)
self.assertEqual(ch_2.multiple_by_two([5, 6, 7], 2), 2)
if __name__ == '__main__':
unittest.main()
|
