def repository():
repo = MemoryRepository()
user1 = User("Myles Kennedy", "123")
user2 = User("Mark Tremonti", "456")
director1 = Director("Taika Waititi")
director2 = Director("Steven Speilberg")
actor1 = Actor("Chris Pratt")
actor2 = Actor("Emma Watson")
actor3 = Actor("Tom Hanks")
genre1 = Genre("Comedy")
genre2 = Genre("Animation")
movie1 = Movie("Moana", 2016)
movie2 = Movie("Ice Age", 2002)
movie1.director = director1
movie2.director = director2
movie1.add_actor(actor1)
movie2.add_actor(actor2)
movie2.add_actor(actor3)
actor2.add_actor_colleague(actor3)
actor3.add_actor_colleague(actor2)
movie1.add_genre(genre1)
movie2.add_genre(genre2)
review1 = Review(movie1, "very nice", 9)
review2 = Review(movie2, "incredible", 10)
user1.add_review(review1)
user2.add_review(review2)
repo.add_movie(movie1)
repo.add_movie(movie2)
repo.add_director(director1)
repo.add_director(director2)
repo.add_actor(actor1)
repo.add_actor(actor2)
repo.add_genre(genre1)
repo.add_genre(genre2)
repo.add_review(review1)
repo.add_review(review2)
repo.add_user(user1)
repo.add_user(user2)
return repo
def test_false_2_check_if_this_actor_worked_with():
actor1 = Actor("Steve Carell")
actor2 = Actor("Amy Adams")
actor3 = Actor("Jack Black")
actor1.add_actor_colleague(actor2)
assert actor1.check_if_this_actor_worked_with(actor3) is False
def test_check_if_this_actor_worked_with():
actor1 = Actor("Steve Carell")
actor2 = Actor("Amy Adams")
actor1.add_actor_colleague(actor2)
assert actor1.check_if_this_actor_worked_with(actor2) is True
def test_add_actor_colleague():
actor1 = Actor("Steve Carell")
actor2 = Actor("Amy Adams")
actor1.add_actor_colleague(actor2)
assert actor1.return_actor_colleague() == [actor2]