#!/usr/bin/python

import unittest

import main



class ExamplesPart1(unittest.TestCase):
    def test_example1(self):
        boxes = [(2, 3, 4)]
        self.assertEqual(main.solve_part1(boxes), 58)

    def test_example2(self):
        boxes = [(1, 1, 10)]
        self.assertEqual(main.solve_part1(boxes), 43)



class ExamplesPart2(unittest.TestCase):
    def test_example1(self):
        boxes = [(2, 3, 4)]
        self.assertEqual(main.solve_part2(boxes), 34)

    def test_example2(self):
        boxes = [(1, 1, 10)]
        self.assertEqual(main.solve_part2(boxes), 14)



if __name__ == "__main__":
    unittest.main()