diff options
Diffstat (limited to '15/test.py')
-rwxr-xr-x | 15/test.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/15/test.py b/15/test.py new file mode 100755 index 0000000..fb71ad8 --- /dev/null +++ b/15/test.py @@ -0,0 +1,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() |