import unittest
class Test01Ogrevalna(unittest.TestCase):
    def test_prestej(self):
        self.assertDictEqual(prestej(["avto", "lok", "avto", "avto", "bomboni", "lok"]), {"avto": 3, "lok": 2, "bomboni": 1})
        self.assertDictEqual(prestej(["avto", "avto", "avto", "avto"]), {"avto": 4})
        self.assertDictEqual(prestej([]), {})


class Test02Obvezna(unittest.TestCase):
    def test_sestej(self):
        self.assertDictEqual(
            sestej([("avto", 2), ("lok", 2), ("avto", 4), ("avto", 1),
                    ("bomboni", 42), ("lok", 3)]),
            {"avto": 7, "lok": 5, "bomboni": 42})
        self.assertDictEqual(sestej([("avto", 1), ("avto", 4), ("avto", 2), ("avto", 1)]), {"avto": 8})
        self.assertDictEqual(sestej([("avto", 1)]), {"avto": 1})
        self.assertDictEqual(sestej([("avto", 2)]), {"avto": 2})
        self.assertDictEqual(sestej([]), {})

    def test_cena(self):
        self.assertEqual(
                cena({"avto": 7, "lok": 5, "bomboni": 42},
                     {"avto": 3, "lok": 5, "bomboni": 1, "dron": 10}),
                88)
        self.assertEqual(
                cena({"avto": 7, "lok": 5, "bomboni": 42},
                     {"avto": 4, "lok": 5, "bomboni": 1, "dron": 10}),
                95)
        self.assertEqual(
                cena({"avto": 7},
                     {"avto": 3, "lok": 5, "bomboni": 1, "dron": 10}),
                21)
        self.assertEqual(cena({}, {"avto": 3, "lok": 5, "bomboni": 1, "dron": 10}), 0)
        self.assertEqual(
                cena({"dron": 2}, {"avto": 3, "lok": 5, "bomboni": 1, "dron": 10}), 20)

    def test_stevilo_daril(self):
        self.assertEqual(stevilo_daril({"avto": 7, "lok": 5, "bomboni": 42}), 54)
        self.assertEqual(stevilo_daril({"avto": 7}), 7)
        self.assertEqual(stevilo_daril({}), 0)

    def test_razlicnih_daril(self):
        self.assertEqual(razlicnih_daril({"avto": 7, "lok": 5, "bomboni": 42}), 3)
        self.assertEqual(razlicnih_daril({"avto": 7}), 1)
        self.assertEqual(razlicnih_daril({}), 0)


class Test03Dodatna(unittest.TestCase):
    def test_napredek(self):
        self.assertDictEqual(
                napredek({"avto": 3, "lok": 5, "bomboni": 1, "dron": 10},
                         {"avto": 8, "lok": 7, "bomboni": 3, "dron": 14}),
                {"avto": 5, "lok": 2, "bomboni": 2, "dron": 4})
        self.assertDictEqual(
                napredek({"avto": 3, "lok": 5, "bomboni": 1, "dron": 10},
                         {"avto": 8, "lok": 5, "bomboni": 3, "dron": 14}),
                {"avto": 5, "bomboni": 2, "dron": 4})
        self.assertDictEqual(
                napredek({"avto": 3, "lok": 5, "bomboni": 1, "dron": 10},
                         {"avto": 8, "lok": 3, "bomboni": 3, "dron": 14}),
                {"avto": 5, "lok": -2, "bomboni": 2, "dron": 4})
        self.assertDictEqual(
                napredek({"avto": 3, "dron": 10},
                         {"avto": 8, "lok": 3, "bomboni": 3, "dron": 14}),
                {"avto": 5, "lok": 3, "bomboni": 3, "dron": 4})
        self.assertDictEqual(
                napredek({"avto": 3, "lok": 5, "bomboni": 1, "dron": 10},
                         {"avto": 8, "dron": 14}),
                {"avto": 5, "lok": -5, "bomboni": -1, "dron": 4})
        self.assertDictEqual(
                napredek({}, {"avto": 8, "dron": 14}),
                {"avto": 8, "dron": 14})
        self.assertDictEqual(
                napredek({"lizika": 4}, {"avto": 8, "dron": 14}),
                {"avto": 8, "dron": 14, "lizika": -4})
        self.assertDictEqual(
                napredek({"lizika": 4}, {}),
                {"lizika": -4})
        self.assertDictEqual(napredek({}, {}), {})


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