def test_given(watchlist): watchlist = WatchList() assert watchlist.size() == 0 watchlist.add_movie(Movie("Moana", 2016)) watchlist.add_movie(Movie("Ice Age", 2002)) watchlist.add_movie(Movie("Guardians of the Galaxy", 2012)) assert watchlist.first_movie_in_watchlist() == Movie("Moana", 2016)
def test_remove_movie(watchlist): watchlist = WatchList() watchlist.add_movie(Movie("Moana", 2016)) watchlist.remove_movie(Movie("Moana", 2016)) assert watchlist.size() == 0
def test_remove_moive_not_in_list(watchlist): watchlist = WatchList() watchlist.add_movie(Movie("Moana", 2016)) watchlist.remove_movie(Movie("Ice Age", 2002)) assert watchlist.size() == 1
def test_size(watchlist): watchlist = WatchList() watchlist.add_movie(Movie("Moana", 2016)) watchlist.add_movie(Movie("Ice Age", 2002)) watchlist.add_movie(Movie("Guardians of the Galaxy", 2012)) assert watchlist.size() == 3