def test_movie_remove_genre(): movie = Movie("Imception", 2011) genre = Genre("Weird") movie.add_actor(genre) assert len(movie.actors) == 1 movie.remove_actor(genre) assert len(movie.actors) == 0
def test_movie_remove_actor(): movie = Movie("Imception", 2011) actor = Actor("John Doe") movie.add_actor(actor) assert len(movie.actors) == 1 movie.remove_actor(actor) assert len(movie.actors) == 0