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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
#!/usr/bin/python
import unittest
import main
class ExamplesPart1(unittest.TestCase):
def setUp(self):
self.lines = [
"123 -> x",
"456 -> y",
"x AND y -> d",
"x OR y -> e",
"x LSHIFT 2 -> f",
"y RSHIFT 2 -> g",
"NOT x -> h",
"NOT y -> i"
]
def test_example1(self):
lines = "\n".join(self.lines).replace("d", "a").splitlines()
self.assertEqual(main.solve_partn(1, lines), 72)
def test_example2(self):
lines = "\n".join(self.lines).replace("e", "a").splitlines()
self.assertEqual(main.solve_partn(1, lines), 507)
def test_example3(self):
lines = "\n".join(self.lines).replace("f", "a").splitlines()
self.assertEqual(main.solve_partn(1, lines), 492)
def test_example4(self):
lines = "\n".join(self.lines).replace("g", "a").splitlines()
self.assertEqual(main.solve_partn(1, lines), 114)
def test_example5(self):
lines = "\n".join(self.lines).replace("h", "a").splitlines()
self.assertEqual(main.solve_partn(1, lines), 65412)
def test_example6(self):
lines = "\n".join(self.lines).replace("i", "a").splitlines()
self.assertEqual(main.solve_partn(1, lines), 65079)
def test_example7(self):
lines = "\n".join(self.lines).replace("x", "a").splitlines()
self.assertEqual(main.solve_partn(1, lines), 123)
def test_example8(self):
lines = "\n".join(self.lines).replace("y", "a").splitlines()
self.assertEqual(main.solve_partn(1, lines), 456)
if __name__ == "__main__":
unittest.main()
|
