#!/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()