import unittest

class Test_ZenskVRodbini(unittest.TestCase):
    def test(self):
        self.assertEqual(zensk_v_rodbini("Friderik II."), 1)
        self.assertEqual(zensk_v_rodbini("Ulrik I."), 1)
        self.assertEqual(zensk_v_rodbini("Ludvik"), 0)
        self.assertEqual(zensk_v_rodbini("Friderik I."), 7)
        self.assertEqual(zensk_v_rodbini("Katarina"), 1)


class Test_Kolikokrat_Ime(unittest.TestCase):
    def test(self):
        self.assertEqual(kolikokrat_ime("Friderik I.", "Friderik"), 3)
        self.assertEqual(kolikokrat_ime("Friderik II.", "Friderik"), 2)
        self.assertEqual(kolikokrat_ime("Barbara", "Friderik"), 0)
        self.assertEqual(kolikokrat_ime("Barbara", "Barbara"), 1)
        self.assertEqual(kolikokrat_ime("Herman I.", "Barbara"), 1)
        self.assertEqual(kolikokrat_ime("Herman I.", "Francelj"), 0)


class Test_VsePotomstvo(unittest.TestCase):
    def test(self):
        self.assertEqual(vse_potomstvo("Friderik II."),
                         {"Friderik III.", "Ulrik II.", "Herman IV.", "Jurij", "Elizabeta II."})
        self.assertEqual(vse_potomstvo("Ulrik I."), {"Viljem", "Ana Poljska"})
        self.assertEqual(vse_potomstvo("Ludvik"), set())
        self.assertEqual(vse_potomstvo("Herman III."), {"Margareta"})