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
