def test_filter_by_director_normal():
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)
assert movie_file_reader.movies_filtered_by_director == [
Movie("Guardians of the Galaxy", 2014),
Movie("Slither", 2006),
Movie("Super", 2010)
]
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_filter_by_director_illeagal_inputs():
filename = 'datafiles/Data1000Movies.csv'
movie_file_reader = MovieFileCSVReader(filename)
movie_file_reader.read_csv_file()
director = Director("Nier Automata")
movie_file_reader.filter_by_director(director)
assert movie_file_reader.movies_filtered_by_director == []
director_1 = "James Gunn"
movie_file_reader.filter_by_director(director_1)
assert movie_file_reader.movies_filtered_by_director == []
director_2 = 12345
movie_file_reader.filter_by_director(director_2)
assert movie_file_reader.movies_filtered_by_director == []
