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():
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 test_watchlist_first_movie_valid(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.first_movie_in_watchlist())
assert watchlist.first_movie_in_watchlist() == Movie("Moana", 2016)
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_not_in_list():
watchlist = WatchList()
assert (f"Size of watchlist: {watchlist.size()}") == f"Size of watchlist: 0"
watchlist.add_movie(Movie("Moana", 2016))
watchlist.add_movie(Movie("Ice Age", 2002))
assert str(watchlist.first_movie_in_watchlist()) == '<Movie Moana, 2016>'
watchlist.remove_movie(Movie("Rogue One", 2016))
assert (f"Size of watchlist: {watchlist.size()}") == f"Size of watchlist: 2"
assert str(watchlist.first_movie_in_watchlist()) == '<Movie Moana, 2016>'
def test_select_and_first():
movie1 = Movie("Moana", 2016)
movie2 = Movie("Booksmart", 2019)
movie3 = Movie("School of Rock", 2003)
movie4 = Movie("Roma", 2018)
watchlist = WatchList()
watchlist.add_movie(movie1)
watchlist.add_movie(movie2)
watchlist.add_movie(movie3)
watchlist.add_movie(movie4)
assert watchlist.select_movie_to_watch(0) == Movie("Moana", 2016)
assert watchlist.select_movie_to_watch(5) is None
assert watchlist.first_movie_in_watchlist() == Movie("Moana", 2016)
watchlist2 = WatchList()
assert watchlist2.first_movie_in_watchlist() is None
def test_create_watch_list():
watchlist = WatchList()
assert (f"Size of watchlist: {watchlist.size()}") == f"Size of watchlist: 0"
watchlist.add_movie(Movie("Moana", 2016))
watchlist.add_movie(Movie("Ice Age", 2002))
watchlist.add_movie(Movie("Guardians of the Galaxy", 2012))
assert str(watchlist.first_movie_in_watchlist()) == '<Movie Moana, 2016>'
def test_first_movie_in_watchlist():
movie1 = Movie("Moana", 2016)
movie2 = Movie("Mulan", 2020)
watchlist1 = WatchList()
watchlist1.add_movie(movie1)
watchlist1.add_movie(movie2)
assert watchlist1.first_movie_in_watchlist() == Movie("Moana", 2016)
def test_init_1(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.first_movie_in_watchlist()
assert (the_movie == movie1) == True
def test_watchlist_first_test(self): #add movie
watchlist = WatchList()
print(f"Size of watchlist: {watchlist.size()}")
#assert watchlist.size() == 0
watchlist.add_movie(Movie("Moana", 2016))
watchlist.add_movie(Movie("Ice Age", 2002))
watchlist.add_movie(Movie("Guardians of the Galaxy", 2012))
print(watchlist.first_movie_in_watchlist())
def test_first_movie_in_watchlist(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.first_movie_in_watchlist() == movie1
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_check_first_movie():
watchlist = WatchList()
assert watchlist.first_movie_in_watchlist() == None
watchlist.add_movie(Movie("Moana", 2016))
watchlist.add_movie(Movie("Ice Age", 2002))
assert watchlist.first_movie_in_watchlist() == Movie("Moana", 2016)
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_same_movie():
watchlist3 = WatchList()
watchlist3.add_movie(Movie("Moana", 2016))
watchlist3.add_movie(Movie("Moana", 2016))
assert repr(watchlist3.first_movie_in_watchlist()) == "<Movie Moana, 2016>"
def test_first_movie_in_watchlist_empty():
watchlist1 = WatchList()
assert watchlist1.first_movie_in_watchlist() == None
def test_watchlist_first_movie_invalid(self):
watchlist = WatchList()
print(watchlist.first_movie_in_watchlist())
assert watchlist.first_movie_in_watchlist() == None
