def test_filter_by_list_of_genres():
filename = 'datafiles/Data1000Movies.csv'
movie_file_reader = MovieFileCSVReader(filename)
movie_file_reader.read_csv_file()
genre = [Genre("Action"), Genre("Crime"), Genre("Mystery")]
movie_file_reader.filter_by_genre(genre)
assert len(movie_file_reader.movies_filtered_by_genre) == 3
assert movie_file_reader.movies_filtered_by_genre == [
Movie("Salt", 2010),
Movie("Jack Reacher", 2012),
Movie("Escape Plan", 2013)
]
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_filter_by_a_genre():
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)
assert len(movie_file_reader.movies_filtered_by_genre) == 13
assert movie_file_reader.movies_filtered_by_genre == [
Movie("Inglourious Basterds", 2009),
Movie("300", 2006),
Movie("Mine", 2016),
Movie("Billy Lynn's Long Halftime Walk", 2016),
Movie("Pan's Labyrinth", 2006),
Movie("Fury", 2014),
Movie("Macbeth", 2015),
Movie("The Boy in the Striped Pyjamas", 2008),
Movie("Frantz", 2016),
Movie("Incendies", 2010),
Movie("Eye in the Sky", 2015),
Movie("Suite Française", 2014),
Movie("Rambo", 2008)
]
def test_filter_by_genres_illeage_inputs():
filename = 'datafiles/Data1000Movies.csv'
movie_file_reader = MovieFileCSVReader(filename)
movie_file_reader.read_csv_file()
genre = Genre("Good")
movie_file_reader.filter_by_genre(genre)
assert movie_file_reader.movies_filtered_by_genre == []
genre = [Genre("Happy")]
movie_file_reader.filter_by_genre(genre)
assert movie_file_reader.movies_filtered_by_genre == []
genre = 123456
movie_file_reader.filter_by_genre(genre)
assert movie_file_reader.movies_filtered_by_genre == []
