otroci = { "Adam": ["Matjaž", "Cilka", "Daniel", "Erik"], "Aleksander": [], "Alenka": [], "Barbara": [], "Cilka": [], "Daniel": ["Elizabeta", "Hans"], "Erik": [], "Elizabeta": ["Ludvik", "Jurij", "Barbara", "Herman"], "Franc": [], "Herman": ["Margareta"], "Hans": [], "Jožef": ["Alenka", "Petra", "Aleksander"], "Jurij": ["Franc", "Jožef"], "Ludvik": [], "Margareta": [], "Matjaž": ["Viljem"], "Mihael": [], "Petra": [], "Tadeja": [], "Viljem": ["Tadeja"], } import unittest class TestObvezna(unittest.TestCase): def test_brez_potomcev(self): self.assertEqual(brez_potomcev("Adam"), 11) self.assertEqual(brez_potomcev("Matjaž"), 1) self.assertEqual(brez_potomcev("Jurij"), 4) class TestDodatna(unittest.TestCase): def test_prvorojenci(self): self.assertEqual(prvorojenci("Tadeja"), set()) self.assertEqual(prvorojenci("Viljem"), {"Tadeja"}) self.assertEqual(prvorojenci("Matjaž"), {"Tadeja", "Viljem"}) self.assertEqual(prvorojenci("Jožef"), {"Alenka"}) self.assertEqual(prvorojenci("Jurij"), {"Franc", "Alenka"}) self.assertEqual(prvorojenci("Elizabeta"), {"Ludvik", "Franc", "Alenka", "Margareta"}) self.assertEqual(prvorojenci("Daniel"), {"Elizabeta", "Ludvik", "Franc", "Alenka", "Margareta"}) self.assertEqual(prvorojenci("Adam"), {"Viljem", "Matjaž", "Tadeja", "Elizabeta", "Ludvik", "Franc", "Alenka", "Margareta"}) if __name__ == "__main__": unittest.main()