import unittest

class Rodbina:
    __getitem__ = {
        "Adam": ["Matjaz", "Cilka", "Daniel", "Erik"],
        "Aleksander": [],
        "Alenka": [],
        "Barbara": [],
        "Cilka": [],
        "Daniel": ["Elizabeta", "Hans"],
        "Erik": [],
        "Elizabeta": ["Ludvik", "Jurij", "Barbara", "Herman"],
        "Franc": [],
        "Herman": ["Margareta"],
        "Hans": [],
        "Jozef": ["Alenka", "Petra", "Aleksander"],
        "Jurij": ["Franc", "Jozef"],
        "Ludvik": [],
        "Margareta": [],
        "Matjaz": ["Viljem"],
        "Mihael": [],
        "Petra": [],
        "Tadeja": [],
        "Viljem": ["Tadeja"],
    }.__getitem__

otroci = Rodbina()

class Testi(unittest.TestCase):
    def test_rodbina(self):
        self.assertSetEqual(
            rodbina("Jurij"),
            {"Jurij", "Franc", "Jozef", "Alenka", "Petra", "Aleksander"})

        self.assertSetEqual(
            rodbina("Elizabeta"),
            {"Elizabeta", "Ludvik",
             "Jurij", "Franc", "Jozef", "Alenka", "Petra", "Aleksander",
             "Barbara", "Herman", "Margareta"
             })

        self.assertSetEqual(
            rodbina("Jozef"),
            {"Jozef", "Alenka", "Petra", "Aleksander"})

        self.assertSetEqual(
            rodbina("Matjaz"),
            {"Matjaz", "Viljem", "Tadeja"})

        self.assertSetEqual(
            rodbina("Tadeja"),
            {"Tadeja"})

    def test_imajo_potomce(self):
        self.assertSetEqual(
            imajo_potomce("Jurij"), {"Jurij", "Jozef"})

        self.assertSetEqual(
            imajo_potomce("Elizabeta"), {"Elizabeta", "Jurij", "Jozef", "Herman"})

        self.assertSetEqual(
            imajo_potomce("Jozef"), {"Jozef"})

        self.assertSetEqual(
            imajo_potomce("Matjaz"), {"Matjaz", "Viljem"})

        self.assertSetEqual(
            imajo_potomce("Tadeja"), set())

    def test_velikost_rodbin(self):
        self.assertDictEqual(
            velikosti_rodbin("Jurij"),
            {"Jurij": 6, "Franc": 1, "Jozef": 4,
             "Alenka": 1, "Petra": 1, "Aleksander": 1})

        self.assertDictEqual(
            velikosti_rodbin("Elizabeta"),
            {"Elizabeta": 11, "Ludvik": 1,
             "Jurij": 6, "Franc": 1, "Jozef": 4,
             "Alenka": 1, "Petra": 1, "Aleksander": 1,
             "Barbara": 1, "Herman": 2, "Margareta": 1
             })

        self.assertDictEqual(
            velikosti_rodbin("Jozef"),
            {"Jozef": 4, "Alenka": 1, "Petra": 1, "Aleksander": 1})

        self.assertDictEqual(
            velikosti_rodbin("Matjaz"),
            {"Matjaz": 3, "Viljem": 2, "Tadeja": 1})

        self.assertDictEqual(
            velikosti_rodbin("Tadeja"),
            {"Tadeja": 1})


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