import unittest


class Test01CesaNi(unittest.TestCase):
    def test_cesa_ni(self):
        narocila = ["klobasa", "riba", "klobasa", "riba", "pizza", "riba", "riba"]
        cesa_ni(narocila, {"klobasa": 10, "riba": 10, "pizza": 10})
        self.assertEqual(narocila, ["klobasa", "riba", "klobasa", "riba", "pizza", "riba", "riba"])

        narocila = ["klobasa", "riba", "klobasa", "riba", "pizza", "riba", "riba"]
        cesa_ni(narocila, {"klobasa": 10, "riba": 3, "pizza": 10})
        self.assertEqual(narocila, ["klobasa", "riba", "klobasa", "riba", "pizza", "riba"])

        narocila = ["klobasa", "riba", "klobasa", "riba", "pizza", "riba", "riba"]
        cesa_ni(narocila, {"klobasa": 10, "riba": 3, "pizza": 0})
        self.assertEqual(narocila, ["klobasa", "riba", "klobasa", "riba", "riba"])

        narocila = ["klobasa", "riba", "klobasa", "riba", "pizza", "riba", "riba"]
        cesa_ni(narocila, {"klobasa": 10, "riba": 3})
        self.assertEqual(narocila, ["klobasa", "riba", "klobasa", "riba", "riba"])

        narocila = ["klobasa", "riba", "klobasa", "riba", "pizza", "riba", "riba"]
        cesa_ni(narocila, {"klobasa": 10, "riba": 3})
        self.assertEqual(narocila, ["klobasa", "riba", "klobasa", "riba", "riba"])

        narocila = ["klobasa", "riba", "klobasa", "riba", "pizza", "riba", "riba"]
        cesa_ni(narocila, {"pizza": 3})
        self.assertEqual(narocila, ["pizza"])

        narocila = ["klobasa", "riba", "klobasa", "riba", "pizza", "riba", "riba"]
        cesa_ni(narocila, {})
        self.assertEqual(narocila, [])

        narocila = []
        cesa_ni(narocila, {"klobasa": 10, "riba": 3, "pizza": 0})
        self.assertEqual(narocila, [])

        narocila = ["c", "a", "a", "c", "a", "b"]
        cesa_ni(narocila, {"a": 1, "c": 2})
        self.assertEqual(narocila, ["c", "a", "c"])

        narocila = ["c", "a", "a", "c", "a", "b"]
        cesa_ni(narocila, {"a": 2, "c": 2})
        self.assertEqual(narocila, ["c", "a", "a", "c"])


class Test02KdoKaj(unittest.TestCase):
    def test_kdo_kaj(self):
        self.assertEqual(
            kdo_kaj({"Ana": "klobasa", "Berta": "klobasa", "Cilka": "pizza"}),
            {"klobasa": {"Ana", "Berta"},
             "pizza": {"Cilka"}})


class Test03Izbrano(unittest.TestCase):
    def test_neizbrano(self):
        jedi = ["klobasa", "zrezek", "riba", "pizza"]
        self.assertEqual(neizbrano(
            jedi, {"Ana": "klobasa", "Berta": "klobasa", "Cilka": "pizza"}),
            ["zrezek", "riba"])
        self.assertEqual(neizbrano(
            jedi, {"Ana": "riba", "Berta": "pizza", "Cilka": "pizza"}),
            ["klobasa", "zrezek"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "riba", "Ema": "klobasa", "Cilka": "pizza"}),
            ["zrezek"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "riba", "Ema": "klobasa", "Cilka": "pizza", "Fanči": "zrezek"}),
            [])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "klobasa", "Ema": "klobasa", "Cilka": "klobasa", "Fanči": "klobasa"}),
            ["zrezek", "riba", "pizza"])

        jedi = ["a", "b", "c"]
        self.assertEqual(neizbrano(
            jedi, {"Dani": "a", "Ema": "a", "Cilka": "a", "Fanči": "a"}),
            ["b", "c"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "a", "Ema": "a", "Cilka": "a", "Fanči": "b"}),
            ["c"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "a", "Ema": "a", "Cilka": "a", "Fanči": "c"}),
            ["b"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "b", "Ema": "b", "Cilka": "b", "Fanči": "b"}),
            ["a", "c"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "b", "Ema": "b", "Cilka": "b", "Fanči": "a"}),
            ["c"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "b", "Ema": "b", "Cilka": "b", "Fanči": "c"}),
            ["a"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "c", "Ema": "c", "Cilka": "c", "Fanči": "c"}),
            ["a", "b"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "c", "Ema": "c", "Cilka": "b", "Fanči": "c"}),
            ["a"])
        self.assertEqual(neizbrano(
            jedi, {"Dani": "c", "Ema": "c", "Cilka": "b", "Fanči": "a"}),
            [])


class Test04(unittest.TestCase):
    def test_prestej(self):
        self.assertEqual(prestej_narocila(
            ["pizza", "klobasa", "riba", "klobasa", "pizza", "klobasa"], "klobasa"
            ), 3)

        self.assertEqual(prestej_narocila(
            ["pizza", "klobasa", "riba", "klobasa", "pizza", "klobasa"], "pizza"
            ), 2)

        self.assertEqual(prestej_narocila(
            ["pizza", "klobasa", "riba", "klobasa", "pizza", "klobasa"], "riba"
            ), 1)

        self.assertEqual(prestej_narocila(
            ["pizza", "klobasa", "riba", "klobasa", "pizza", "klobasa"], "zrezek"
            ), 0)


class Test05(unittest.TestCase):
    def test_natakar(self):
        n = Natakar()

        self.assertEqual(n.narocila(), {})
        n.naroci("Ana", "pizza")
        self.assertEqual(n.narocila(), {"Ana": {"pizza"}})

        n.naroci("Berta", "klobasa")
        self.assertEqual(n.narocila(), {"Ana": {"pizza"}, "Berta": {"klobasa"}})

        n.naroci("Ana", "pivo")
        self.assertEqual(n.narocila(), {"Ana": {"pizza", "pivo"}, "Berta": {"klobasa"}})

        n.naroci("Berta", "pivo")
        self.assertEqual(n.narocila(), {"Ana": {"pizza", "pivo"}, "Berta": {"klobasa", "pivo"}})
        self.assertEqual(n.cakajoce_osebe(), {"Ana", "Berta"})
        self.assertEqual(n.kdo_je_narocil("pivo"), {"Ana", "Berta"})
        self.assertEqual(n.kdo_je_narocil("pizza"), {"Ana"})
        self.assertEqual(n.kdo_je_narocil("klobasa"), {"Berta"})
        self.assertEqual(n.kdo_je_narocil("zrezek"), set())

        n.prinesi("Ana", "pizza")
        self.assertEqual(n.narocila(), {"Ana": {"pivo"}, "Berta": {"klobasa", "pivo"}})
        self.assertEqual(n.cakajoce_osebe(), {"Ana", "Berta"})
        self.assertEqual(n.kdo_je_narocil("pivo"), {"Ana", "Berta"})
        self.assertEqual(n.kdo_je_narocil("pizza"), set())
        self.assertEqual(n.kdo_je_narocil("klobasa"), {"Berta"})
        self.assertEqual(n.kdo_je_narocil("zrezek"), set())

        n.prinesi("Berta", "pivo")
        self.assertEqual(n.narocila(), {"Ana": {"pivo"}, "Berta": {"klobasa"}})
        self.assertEqual(n.cakajoce_osebe(), {"Ana", "Berta"})
        self.assertEqual(n.kdo_je_narocil("pivo"), {"Ana"})
        self.assertEqual(n.kdo_je_narocil("pizza"), set())
        self.assertEqual(n.kdo_je_narocil("klobasa"), {"Berta"})
        self.assertEqual(n.kdo_je_narocil("zrezek"), set())

        n.prinesi("Ana", "pivo")
        self.assertEqual(n.narocila(), {"Berta": {"klobasa"}})
        self.assertEqual(n.cakajoce_osebe(), {"Berta"})
        self.assertEqual(n.kdo_je_narocil("pivo"), set())
        self.assertEqual(n.kdo_je_narocil("pizza"), set())
        self.assertEqual(n.kdo_je_narocil("klobasa"), {"Berta"})
        self.assertEqual(n.kdo_je_narocil("zrezek"), set())

        n.prinesi("Berta", "klobasa")
        self.assertEqual(n.narocila(), {})
        self.assertEqual(n.cakajoce_osebe(), set())
        self.assertEqual(n.kdo_je_narocil("pivo"), set())
        self.assertEqual(n.kdo_je_narocil("pizza"), set())
        self.assertEqual(n.kdo_je_narocil("klobasa"), set())
        self.assertEqual(n.kdo_je_narocil("zrezek"), set())


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