def test_add_colleague(): actor1 = Actor("Angelina Jolie") actor2 = Actor("Brad Pitt") assert actor1.check_if_this_actor_worked_with(actor2) == False 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
def test_actor_check_if_this_actor_worked_with(): actor_1 = Actor("Angelina Jolie") actor_2 = Actor("Tom Cruise") actor_3 = Actor("Chris Pratt") actor_1.add_actor_colleague(actor_2) assert actor_1.check_if_this_actor_worked_with(actor_2) is True assert actor_1.check_if_this_actor_worked_with(actor_3) is False