Python WatchList.select_movie_to_watch Examples

Example #1

File: test_domain_model.py Project: r-ajpreet/CS235Flix

def test_watchlist_select_movie_to_watch():
    watchlist = WatchList()
    movie1 = Movie("Moana", 2016)
    movie2 = Movie("Frozen", 2014)
    assert watchlist.select_movie_to_watch(0) is None
    watchlist.add_movie(movie1)
    assert watchlist.select_movie_to_watch(0) == movie1
    assert watchlist.select_movie_to_watch(1) is None
    watchlist.add_movie(movie2)
    assert watchlist.select_movie_to_watch(1) == movie2
    watchlist.remove_movie(movie1)
    assert watchlist.select_movie_to_watch(1) is None
    assert watchlist.select_movie_to_watch(0) == movie2
    assert watchlist.select_movie_to_watch("0") is None
    assert watchlist.select_movie_to_watch(-1) is None

Example #2

def test_watch_list_select_movie_out_of_range():
    watchlist8 = WatchList()
    watchlist8.add_movie(Movie("Moana", 2016))
    watchlist8.add_movie(Movie("Transformers", 2007))
    watchlist8.add_movie(Movie("Ice Age", 2002))
    assert repr(watchlist8.select_movie_to_watch(3)) == "None"

Example #3

def test_watch_list_select_movie():
    watchlist7 = WatchList()
    watchlist7.add_movie(Movie("Moana", 2016))
    watchlist7.add_movie(Movie("Transformers", 2007))
    watchlist7.add_movie(Movie("Ice Age", 2002))
    assert repr(watchlist7.select_movie_to_watch(2)) == "<Movie Ice Age, 2002>"