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
|
#!/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.longest_parenthesis('(()())'), 6)
self.assertEqual(ch_1.longest_parenthesis(')()())'), 4)
self.assertEqual(ch_1.longest_parenthesis('((()))()(((()'), 8)
self.assertEqual(ch_1.longest_parenthesis('))))((()('), 2)
self.assertEqual(ch_1.longest_parenthesis('()(()'), 2)
def test_ch_2(self):
self.assertEqual(ch_2.magic_expression(123, 6), ["1+2+3", "1*2*3"])
self.assertEqual(ch_2.magic_expression(105, 5), ["10-5", "1*0+5"])
self.assertEqual(ch_2.magic_expression(232, 8), ["2+3*2", "2*3+2"])
self.assertEqual(ch_2.magic_expression(1234, 10), ["1+2+3+4", "1*2*3+4"])
self.assertEqual(
ch_2.magic_expression(1001, 2),
['1+0+0+1', '1+0-0+1', '1+0*0+1', '1-0+0+1', '1-0-0+1', '1-0*0+1']
)
if __name__ == '__main__':
unittest.main()
|
