def test_remove_genre():
movie = Movie("Moana", 2016)
genres = [
Genre("Animation"),
Genre("Action"),
Genre("Fantasy"),
Genre("Children's film"),
Genre("Adventure")
]
for genre in genres:
movie.add_genre(genre)
assert repr(
movie.genres
) == "[<Genre Animation>, <Genre Action>, <Genre Fantasy>, <Genre Children's film>, <Genre Adventure>]"
movie.remove_genre(Genre("Adventure"))
assert repr(
movie.genres
) == "[<Genre Animation>, <Genre Action>, <Genre Fantasy>, <Genre Children's film>]"
movie.remove_genre(Genre("Adventure"))
assert repr(
movie.genres
) == "[<Genre Animation>, <Genre Action>, <Genre Fantasy>, <Genre Children's film>]"
def test_movie_remove_genre():
movie1 = Movie("Moana", 2016)
movie1.add_genre(Genre("Animation"))
# correct input
movie1.remove_genre(Genre("Animation"))
assert movie1.genres == []
# incorrect input
movie1.add_genre(Genre("Animation"))
movie1.remove_genre(Director("Animation"))
assert movie1.genres == [Genre("Animation")]
# removing a genre not in list
movie1.remove_genre(Genre("Comedy"))
assert movie1.genres == [Genre("Animation")]
