def test_movie_remove_genre(): movie1 = Movie("Princess and The Frog", 2009) genre1 = Genre("Animation") genre2 = Genre("Musical") movie1.add_genre(genre1) movie1.add_genre(genre2) movie1.remove_genre(genre1) assert movie1.return_genre() == [genre2]
def test_movie_add_genre(): movie1 = Movie("Princess and The Frog", 2009) genre1 = Genre("Animation") movie1.add_genre(genre1) assert movie1.return_genre() == [genre1]