예제 #1
0
    def test_add_daughter(self):
        husband = Person('Evan', 'Male')
        wife = Person('Diana', 'Female')
        family = Family(husband, wife)
        daughter_name = 'Nisha'
        family.add_daughter(daughter_name)

        assert family.children[0].name == 'Nisha'
        assert family.children[0].gender == 'Female'
예제 #2
0
    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