import unittest


class Test(unittest.TestCase):
    def test_privlacnost(self):
        # "ACG GGG ACG TTG GCA AAG ACG TTG"
        self.assertEqual(3, privlacnost("ACGGGGACGTTGGCAAAGACGTTG", ["GGG", "ACG", "TTT", "TTG"]))
        self.assertEqual(3, privlacnost("ACGGGGACGTTGGCAAAGACGTTG", ["GGG", "ACG", "TTT"]))
        self.assertEqual(2, privlacnost("ACGGGGACGTTGGCAAAGACGTTG", ["GGG", "TTT", "TTG"]))
        self.assertEqual(1, privlacnost("ACGGGGACGTTGGCAAAGACGTTG", ["GGG", "TTT"]))
        self.assertEqual(0, privlacnost("ACGGGGACGTTGGCAAAGACGTTG", ["AAA"]))

        # ACC AGC AGC AGT
        # Če tole vrne 3, potem gledš CAG. To je narobe.
        self.assertEqual(2, privlacnost("ACCAGCAGCAGT" , ["AGC", "CAG"]))

    def test_okuzeni(self):
        self.assertEqual({"Berta", "Dani", "Fanči"},
            okuzeni([("Bzuf", "Ana"),
                     ("Bzzi", "Ana"),
                     ("Zuuum", "Berta"),  # "Zuuum" se okuži
                     ("Bzzi", "Berta"),   # "Bzzi" se okuži
                     ("Bzuf", "Cilka"),
                     ("Bzzi", "Dani"),    # "Bzzi" okuži Dani
                     ("Zzzzaf", "Dani"),  # "Zzzzaf" se okuži
                     ("Bzuf", "Ema"),
                     ("Bzuf", "Fanči"),
                     ("Zzzzaf", "Fanči")  # Fanči se okuži
                     ],
                    "Berta"))

        self.assertEqual({"Francl"},
            okuzeni([("Bzuf", "Ana"),
                     ("Bzzi", "Ana"),
                     ("Zuuum", "Berta"),
                     ("Bzzi", "Berta"),
                     ("Bzuf", "Cilka"),
                     ("Bzzi", "Dani"),
                     ("Zzzzaf", "Dani"),
                     ("Bzuf", "Ema"),
                     ("Bzuf", "Fanči"),
                     ("Zzzzaf", "Fanči")
                     ],
                    "Francl"))

        self.assertEqual({"Francl"}, okuzeni([], "Francl"))

    def test_alergija(self):
        otroci = {
            "Adam": ["Matjaž", "Cilka", "Daniel"],
            "Aleksander": [],
            "Alenka": [],
            "Barbara": [],
            "Cilka": [],
            "Daniel": ["Elizabeta", "Hans"],
            "Erik": [],
            "Elizabeta": ["Ludvik", "Jurij", "Barbara"],
            "Franc": [],
            "Herman": ["Margareta"],
            "Hans": ["Herman", "Erik"],
            "Jožef": ["Alenka", "Aleksander", "Petra"],
            "Jurij": ["Franc", "Jožef"],
            "Ludvik": [],
            "Margareta": [],
            "Matjaž": ["Viljem"],
            "Petra": [],
            "Tadeja": [],
            "Viljem": ["Tadeja"]
        }
        self.assertTrue(
            alergija("Jurij",
                     {"Alenka", "Petra", "Aleksander", "Franc", "Jožef", "Jurij"},
                     otroci))

        self.assertTrue(
            alergija("Jurij",
                     {"Alenka", "Petra", "Aleksander", "Franc", "Jožef", "Jurij", "Adam", "Viljem"},
                     otroci))

        self.assertFalse(
            alergija("Jurij",
                     {"Alenka", "Petra", "Franc", "Jožef", "Jurij"},
                     otroci))

        self.assertFalse(
            alergija("Jurij",
                     {"Alenka", "Petra", "Aleksander", "Franc", "Jožef"},
                     otroci))

        self.assertTrue(
            alergija("Alenka", {"Alenka"}, otroci))

    def test_ogrozenost(self):
        open("podatki_22_06_06.txt", "wt").write("""Kranj 5 2 8 6
Maribor 3 6 1
Ljubljana 2 10 5
NovoMesto 1 2 3
""")
        ogrozenost("podatki_22_06_06.txt", "zbrano.txt")
        self.assertEqual("""NovoMesto               6
Maribor                10
Ljubljana              17
Kranj                  21""", open("zbrano.txt").read().strip())

    def test_komar(self):
        bz = Komar()
        self.assertTrue(bz.ziv())
        bz.cepljenje("asdf")
        bz.cepljenje("qwer")
        bz.cepljenje("uiop")
        self.assertTrue(bz.ziv())
        bz.spricanje("asdf")
        self.assertTrue(bz.ziv())
        bz.spricanje("qwer")
        self.assertTrue(bz.ziv())
        bz.spricanje("uiop")
        self.assertTrue(bz.ziv())
        bz.spricanje("vbnm")
        self.assertFalse(bz.ziv())
        bz.spricanje("asdf")
        self.assertFalse(bz.ziv())


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