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]