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
