def test_watchlist_size(self): watchlist = WatchList() watchlist.add_movie(Movie("Moana", 2016)) watchlist.add_movie(Movie("Ice Age", 2002)) watchlist.add_movie(Movie("Guardians of the Galaxy", 2012)) print(watchlist.size()) assert watchlist.size() == 3
def test_add_movie(): w = WatchList() w.add_movie(movie) assert w.first_movie_in_watchlist() == movie assert w.size() == 1 w.add_movie(movie2) assert w.first_movie_in_watchlist() == movie assert w.size() == 2
def test_remove_movie(): w = WatchList() w.add_movie(movie) assert w.size() == 1 w.remove_movie(movie2) assert w.size() == 1 w.remove_movie(movie) assert w.size() == 0
def test_remove_movie(self): watchlist = WatchList() movie1 = Movie("Star Wars", 1999) watchlist.add_movie(movie1) assert watchlist.size() == 1 watchlist.remove_movie(movie1) assert watchlist.size() == 0 # tests it removes a movie if it is there watchlist.remove_movie(movie1) assert watchlist.size() == 0 # tests it does nothing if movie not there
def test_size(): watchlist = WatchList() assert watchlist.size() == 0 movie1 = Movie("Moana", 2016) movie2 = Movie("Booksmart", 2019) movie3 = Movie("School of Rock", 2003) movie4 = Movie("Roma", 2018) watchlist.add_movie(movie1) watchlist.add_movie(movie2) watchlist.add_movie(movie3) watchlist.add_movie(movie4) assert watchlist.size() == 4
def test(): watchlist = WatchList() assert watchlist.size() == 0 watchlist.add_movie(Movie("Up", 2009)) assert watchlist.size() == 1 watchlist.add_movie(Movie("Down", 1999)) assert watchlist.size() == 2 watchlist.add_movie(Movie("XYZ", 2013)) assert watchlist.size() == 3 watchlist.add_movie(Movie("Anabelle", 2020)) assert watchlist.size() == 4 watchlist.add_movie(Movie("Anabelle", 2020)) assert watchlist.size() == 4 i = iter(watchlist) repr(next(i)) == "<Movie Up, 2009>" repr(next(i)) == "<Movie Down, 1999>" repr(next(i)) == "<Movie XYZ, 2013>" repr(next(i)) == "<Movie Anabelle, 2020>" with pytest.raises(StopIteration): next(i) watchlist.remove_movie(Movie("Up", 2009)) assert watchlist.size() == 3 watchlist.remove_movie(Movie("Left", 2009)) assert watchlist.size() == 3 assert repr(watchlist.select_movie_to_watch(0)) == "<Movie Down, 1999>" assert watchlist.select_movie_to_watch(4) == None assert repr(watchlist.first_movie_in_watchlist()) == "<Movie Down, 1999>" watchlist.remove_movie(Movie("Down", 1999)) watchlist.remove_movie(Movie("XYZ", 2013)) watchlist.remove_movie(Movie("Anabelle", 2020)) assert watchlist.first_movie_in_watchlist() == None
def constructor_test(): watchlist = WatchList() print(watchlist.size()) watchlist.add_movie(Movie("Up", 2009)) print(watchlist.size()) watchlist.add_movie(Movie("Down", 1999)) print(watchlist.size()) watchlist.add_movie(Movie("XYZ", 2013)) print(watchlist.size()) watchlist.add_movie(Movie("Anabelle", 2020)) print(watchlist.size()) watchlist.add_movie(Movie("Anabelle", 2020)) print(watchlist.size()) i = iter(watchlist) print(next(i)) print(next(i)) print(next(i)) for movie in i: print(movie) watchlist.remove_movie(Movie("Up", 2009)) print(watchlist.size()) watchlist.remove_movie(Movie("Left", 2009)) print(watchlist.size()) print(watchlist.select_movie_to_watch(0)) print(watchlist.select_movie_to_watch(4)) print(watchlist.first_movie_in_watchlist()) watchlist.remove_movie(Movie("Down", 1999)) watchlist.remove_movie(Movie("XYZ", 2013)) watchlist.remove_movie(Movie("Anabelle", 2020)) print(watchlist.first_movie_in_watchlist())
def test_five_random_movie_from_all_movies(): seed = 116 watchlist = WatchList() filename = 'datafiles/Data1000Movies.csv' movie_file_reader = MovieFileCSVReader(filename) movie_file_reader.read_csv_file() movie_list = movie_file_reader.dataset_of_movies random_movie_1 = watchlist.add_random(movie_list, seed) assert random_movie_1 == movie_list[841] random_movie_2 = watchlist.add_random(movie_list, seed) assert random_movie_2 == movie_list[763] random_movie_3 = watchlist.add_random(movie_list, seed) assert random_movie_3 == movie_list[614] random_movie_4 = watchlist.add_random(movie_list, seed) assert random_movie_4 == movie_list[559] random_movie_5 = watchlist.add_random(movie_list, seed) assert random_movie_5 == movie_list[739] assert watchlist.size() == 5
def test_check_size(): movie1 = Movie("Moana", 2016) movie2 = Movie("Mulan", 2020) watchlist1 = WatchList() watchlist1.add_movie(movie1) watchlist1.add_movie(movie2) assert watchlist1.size() == 2
def test_adding_and_removing(): watchlist = WatchList() movie1 = Movie("Moana", 2016) movie2 = Movie("Booksmart", 2019) movie3 = Movie("School of Rock", 2003) movie4 = Movie("Roma", 2018) watchlist.add_movie(movie1) watchlist.add_movie(movie2) watchlist.add_movie(movie3) watchlist.add_movie(movie4) watchlist.add_movie(movie4) assert watchlist.size() == 4 watchlist.remove_movie(movie1) assert watchlist.size() == 3 watchlist.remove_movie(Movie("Juno", 2007)) assert watchlist.size() == 3
def test_remove_movie_which_is_not_in_watchlist(): movie1 = Movie("Moana", 2016) movie2 = Movie("Mulan", 2020) watchlist1 = WatchList() watchlist1.add_movie(movie1) watchlist1.remove_movie(movie2) assert watchlist1.size() == 1
def test_init_3(self): watchlist_1 = WatchList() movie1 = Movie("Moana", 2016) movie2 = Movie("Harry Potter", 2010) watchlist_1.add_movie(movie1) watchlist_1.add_movie(movie2) watchlist_1.add_movie(movie1) assert watchlist_1.size() == 2
def test_init(self): watchlist_1 = WatchList() movie1 = Movie("Moana", 2016) movie2 = Movie("Harry Potter", 2010) watchlist_1.add_movie(movie1) watchlist_1.add_movie(movie2) the_movie = watchlist_1.select_movie_to_watch(1) assert (the_movie == movie2) == True assert watchlist_1.size() == 2
def test_init_2(self): watchlist_1 = WatchList() movie1 = Movie("Moana", 2016) movie2 = Movie("Harry Potter", 2010) watchlist_1.add_movie(movie1) watchlist_1.add_movie(movie2) number = 0 for i in watchlist_1: number += 1 assert number == watchlist_1.size()
def test_watchlist(): watchlist = WatchList() movies = [ Movie("Moana", 2016), Movie("The Lion King", 1994), Movie("Guardians of the Galaxy", 2016), Movie("Prometheus", 2012), Movie("La La Land", 2016) ] for movie in movies: watchlist.add_movie(movie) assert watchlist.size() == 5 assert watchlist.first_movie_in_watchlist() == movies[0] assert watchlist.select_movie_to_watch(2) == movies[2] assert watchlist.select_movie_to_watch(7) is None movies2 = [] for movie in watchlist: movies2.append(movie) assert movies == movies2 watchlist.remove_movie(movies[3]) assert watchlist.size() == 4 watchlist.add_movie(movies[1]) assert watchlist.size() == 4 watchlist_iter = iter(watchlist) assert next(watchlist_iter) == movies[0] assert next(watchlist_iter) == movies[1] assert next(watchlist_iter) == movies[2] assert next(watchlist_iter) == movies[4] with pytest.raises(StopIteration): next(watchlist_iter)
def test_size(self): movie1 = Movie("Moana", 2016) movie2 = Movie("Ice Age", 2002) movie3 = Movie("Wall-E", 2008) watchlist = WatchList() watchlist.add_movie(movie1) watchlist.add_movie(movie2) watchlist.add_movie(movie3) assert watchlist.size() == 3
def test_movie_list_filtered_by_director_duplicates(): seed = 116 watchlist = WatchList() filename = 'datafiles/Data1000Movies.csv' movie_file_reader = MovieFileCSVReader(filename) movie_file_reader.read_csv_file() director = Director("James Gunn") movie_file_reader.filter_by_director(director) movie_list = movie_file_reader.movies_filtered_by_director random_movie_1 = watchlist.add_random(movie_list, seed) assert random_movie_1 == movie_list[2] random_movie_2 = watchlist.add_random(movie_list, seed) assert random_movie_2 == movie_list[1] random_movie_2 = watchlist.add_random(movie_list, seed) assert random_movie_2 == movie_list[0] assert watchlist.size() == 3
def test_movie_list_filtered_by_genres(): seed = 116 watchlist = WatchList() filename = 'datafiles/Data1000Movies.csv' movie_file_reader = MovieFileCSVReader(filename) movie_file_reader.read_csv_file() genre = Genre("War") movie_file_reader.filter_by_genre(genre) movie_list = movie_file_reader.movies_filtered_by_genre random_movie_1 = watchlist.add_random(movie_list, seed) assert random_movie_1 == movie_list[11] random_movie_2 = watchlist.add_random(movie_list, seed) assert random_movie_2 == movie_list[9] random_movie_3 = watchlist.add_random(movie_list, seed) assert random_movie_3 == movie_list[8] assert watchlist.size() == 3
def test_size(self): # this test is a bit silly since most the other tests rely on size working? watchlist = WatchList() assert watchlist.size() == 0 watchlist.remove_movie(Movie("Star Wars", 1999)) assert watchlist.size() == 0 # ^ I believe this is a great test, checking that remove_movie # doesn't subtract 1 from size unless a movie is indeed removed watchlist.add_movie(Movie("Star Wars", 1999)) assert watchlist.size() == 1 watchlist.add_movie(Movie("Ice Age", 2002)) assert watchlist.size() == 2 watchlist.add_movie(Movie("Guardians of the Galaxy", 2012)) assert watchlist.size() == 3 watchlist.remove_movie(Movie("Star Wars", 1999)) # ^ as per note above note, __eq___ is defined by same title and year, so this should do the job of removing assert watchlist.size() == 2 # test size reacts to movies being removed
def test_init_and_size(self): watchlist = WatchList() assert watchlist.size() == 0 # check starts with 0 movies in watchlist (assumes size works)
def test_check__size_empty(): watchlist1 = WatchList() assert watchlist1.size() == 0
def test_add_same_movie_again(): movie1 = Movie("Moana", 2016) watchlist1 = WatchList() watchlist1.add_movie(movie1) watchlist1.add_movie(movie1) assert watchlist1.size() == 1
def test_add_movie(self): watchlist = WatchList() watchlist.add_movie(Movie("Star Wars", 1999)) assert watchlist.size() == 1 # tests that add and size are working.. watchlist.add_movie(Movie("Star Wars", 1999)) assert watchlist.size() == 1 # test that indeed a duplicate movie can't be added
def test_add_movie_twice(): w = WatchList() w.add_movie(movie) assert w.size() == 1 w.add_movie(movie) assert w.size() == 1
def test_remove_movie_not_in_list(): watchlist = WatchList() watchlist.remove_movie(Movie("Your name", 2006)) assert watchlist.size() == 0
def test_remove_movie(): watchlist5 = WatchList() watchlist5.add_movie(Movie("Moana", 2016)) watchlist5.remove_movie(Movie("Transformers", 2007)) assert watchlist5.size() == 1 assert repr(watchlist5.first_movie_in_watchlist()) == "<Movie Moana, 2016>"
def test_add_and_remove_movie(): watchlist6 = WatchList() watchlist6.add_movie(Movie("Moana", 2016)) watchlist6.remove_movie(Movie("Moana", 2016)) assert watchlist6.watch_list == [] assert watchlist6.size() == 0
def test_init(): watchlist1 = WatchList() assert watchlist1.watch_list == [] assert watchlist1.size() == 0
def test_check_size(): watchlist4 = WatchList() watchlist4.add_movie(Movie("Moana", 2016)) watchlist4.add_movie(Movie("Ice Age", 2002)) assert watchlist4.size() == 2
def test_add_same_movie(): watchlist = WatchList() watchlist.add_movie(Movie("Your Name", 2006)) watchlist.add_movie(Movie("Your Name", 2006)) assert watchlist.size() == 1