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_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_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