1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#!/usr/bin/python
import unittest
import main
class Examples(unittest.TestCase):
def test_example1(self):
lines = [
"Butterscotch: capacity -1, durability -2, flavor 6, texture 3, calories 8",
"Cinnamon: capacity 2, durability 3, flavor -2, texture -1, calories 3",
"Dummy: capacity 0, durability 0, flavor 0, texture 0, calories 0",
"Dummy: capacity 0, durability 0, flavor 0, texture 0, calories 0"
]
self.assertEqual(main.solve_partn(1, lines), 62842880)
self.assertEqual(main.solve_partn(2, lines), 57600000)
if __name__ == "__main__":
unittest.main()
|
