def test_check_if_this_actor_worked_with(self): actor1 = Actor("Angelina Jolie") actor2 = Actor("Brad Pitt") actor3 = Actor("Not an Actor") actor1.add_actor_colleague(actor2) assert actor1.check_if_this_actor_worked_with(actor2) == True assert actor2.check_if_this_actor_worked_with(actor1) == True assert actor1.check_if_this_actor_worked_with(actor3) == False
def test_add_actor_colleague(self): actor1 = Actor("Angelina Jolie") actor2 = Actor("Brad Pitt") actor3 = Actor("Not an Actor") actor1.add_actor_colleague(actor2) assert actor1.get_number_of_colleagues() == 1 assert actor2.get_number_of_colleagues() == 1