def test_remove(): watchlist = WatchList() watchlist.remove_movie(Movie("Moana", 2016)) assert watchlist.size() == 0 watchlist.add_movie(Movie("Moana", 2016)) watchlist.remove_movie(Movie("Moana", 2016)) assert watchlist.size() == 0
def test_add(): watchlist = WatchList() watchlist.add_movie("hello world") assert watchlist.size() == 0 watchlist = WatchList() watchlist.add_movie(Movie("Moana", 2016)) assert watchlist.size() == 1
def test_repeated_add(): watchlist = WatchList() watchlist.add_movie(Movie("Moana", 2016)) watchlist.add_movie(Movie("Moana", 2016)) assert watchlist.size() == 1