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
