import unittest

class Test(unittest.TestCase):
    def test_01_obvezna_koliko_cesa(self):
        with self.assertRaises(RecursionError, msg="Funkcija mora biti rekurzivna!"):
            koliko_cesa("kava", ["kava"] * 2000)

        s = ["kava", "kava", "telovadba", "zdravnik", "kava", "zdravnik"]
        self.assertEqual(0, koliko_cesa("kava", []))
        self.assertEqual(1, koliko_cesa("kava", ["kava"]))
        self.assertEqual(0, koliko_cesa("kava", ["zdravnik"]))

        self.assertEqual(3, koliko_cesa("kava", s))
        self.assertEqual(3, koliko_cesa("kava", s))
        self.assertEqual(2, koliko_cesa("zdravnik", s))
        self.assertEqual(0, koliko_cesa("zoo", s))
        self.assertEqual(0, koliko_cesa("zoo", s))
        self.assertEqual(3, koliko_cesa("kava", s))
        self.assertEqual(2, koliko_cesa("zdravnik", s))

    def test_02_obvezna_koliko_aktivnosti(self):
        with self.assertRaises(RecursionError, msg="Funkcija mora biti rekurzivna!"):
            koliko_aktivnosti("kava", [("Ana", "kava")] * 2000)

        s = [("Ana", "kava"), ("Berta", "kava"), ("Ana", "telovadba"),
             ("Cilka", "zdravnik"), ("Cilka", "kava"), ("Dani", "zdravnik")]
        self.assertEqual(0, koliko_aktivnosti("kava", []))
        self.assertEqual(1, koliko_aktivnosti("kava", [("Ana", "kava")]))
        self.assertEqual(0, koliko_aktivnosti("kava", [("Cilka", "zdravnik")]))

        self.assertEqual(3, koliko_aktivnosti("kava", s))
        self.assertEqual(3, koliko_aktivnosti("kava", s))
        self.assertEqual(2, koliko_aktivnosti("zdravnik", s))
        self.assertEqual(0, koliko_aktivnosti("zoo", s))
        self.assertEqual(0, koliko_aktivnosti("zoo", s))
        self.assertEqual(3, koliko_aktivnosti("kava", s))
        self.assertEqual(2, koliko_aktivnosti("zdravnik", s))

    def test_03_okuzena(self):
        okuzene = {"Berta", "Cilka"}
        self.assertFalse(okuzena("Ana", [], okuzene))
        self.assertFalse(okuzena("Ana", [{"Ana", "Dani"}], okuzene))
        self.assertFalse(okuzena("Ana", [{"Ana", "Dani"}, {"Berta", "Dani"}], okuzene))
        self.assertFalse(okuzena("Ana", [{"Ana", "Dani"}, {"Berta", "Ema"}], okuzene))

        self.assertTrue(okuzena("Ana", [{"Ana", "Dani"}, {"Berta", "Ema"}], {"Ana", "Ema"}))

        self.assertTrue(okuzena("Ana", [{"Berta", "Ema", "Ana"}, {"Ema", "Fanči"}, {"Ana", "Greta"}], okuzene))
        self.assertTrue(okuzena("Ana", [{"Ana", "Dani"}, {"Berta", "Ema", "Ana"}], okuzene))


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