class TestFamily(unittest.TestCase): def setUp(self): self.King_root = Person("King", "M") self.family = Family(self.King_root) self.person_1 = Person("person1", "M") self.person_2 = Person("person2", "F") def test_search_functionality(self): with self.assertRaises(custom_exceptions.PersonNotFound): self.family.search_member(self.person_1) def test_child_addition(self): with self.assertRaises(custom_exceptions.ChildAdditionFailed): self.family.add_child_in_family(self.King_root, self.person_1) def test_get_relationship(self): with self.assertRaises(custom_exceptions.CommandNotFound): self.family.get_relationship(self.King_root, "unknown_relation")
Satvy = Person("Satvy", constants.FEMALE) Asva = Person("Asva", constants.MALE) Krpi = Person("Krpi", constants.FEMALE) Vyas = Person("Vyas", constants.MALE) Atya = Person("Atya", constants.FEMALE) Yodhan = Person("Yodhan", constants.MALE) Laki = Person("Laki", constants.MALE) Lavnya = Person("Lavnya", constants.FEMALE) Vasa = Person("Vasa", constants.MALE) Kriya = Person("Kriya", constants.MALE) Krithi = Person("Krithi", constants.FEMALE) #Relating the population family.add_spouse_in_family(King_Shan, Queen_Anga) family.add_child_in_family(Queen_Anga, Chit) family.add_child_in_family(Queen_Anga, Ish) family.add_child_in_family(Queen_Anga, Vich) family.add_child_in_family(Queen_Anga, Aras) family.add_child_in_family(Queen_Anga, Satya) family.add_spouse_in_family(Chit, Amba) family.add_child_in_family(Amba, Dritha) family.add_child_in_family(Amba, Tritha) family.add_child_in_family(Amba, Vritha) family.add_spouse_in_family(Vich, Lika) family.add_child_in_family(Lika, Vila) family.add_child_in_family(Lika, Chika) family.add_spouse_in_family(Aras, Chitra)