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")]