def test_add_son(self): husband = Person('Evan', 'Male') wife = Person('Diana', 'Female') family = Family(husband, wife) son_name = 'Alex' family.add_son(son_name) assert family.children[0].name == 'Alex' assert family.children[0].gender == 'Male'
def test_get_sisters(self): husband = Person('Evan', 'Male') wife = Person('Diana', 'Female') family = Family(husband, wife) family.add_son('John') family.add_son('Alex') family.add_son('Joe') family.add_daughter('Nisha') list_of_sisters = family.get_sisters('Joe') assert 'Nisha' in list_of_sisters assert 'Alex' not in list_of_sisters