예제 #1
0
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
예제 #2
0
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