def reset(self): """ reset the data/clear out the data so more than one search can happen :return: """ self.search = Search() self.movie_title = '' self.imdb_num = '' time.sleep(3) self.main_menu()
def test_init(): search = Search() assert(search.imdb_id == '?i=') # A valid IMDb ID (e.g. tt1285016) assert(search.title == '?t=') # Movie title to search for. assert(search.s_type== '&type=movie') # movie, series, episode Type of result to return. assert(search.year == '&y=') # Year of release. assert(search.response_type == '&r=json') assert(search.plot == '&plot=short') assert(search.tomatoes == '&tomatoes=true') assert(search.url == 'http://www.omdbapi.com/') assert(search.response_list == ['Title', 'Plot', 'Actors', 'Year'])
def test_episode(): search = Search() search.episode('Game+of+Thrones', '2', '1') assert(search.m_title == u'?t=Game+of+Thrones&Season=2&Episode=1') assert(search.url == 'http://www.omdbapi.com/?t=Game+of+Thrones&Season=2&Episode=1&y=&plot=short&tomatoes=true')
def test_imdbid(): search = Search() search.imdbid('tt0079501') assert(search.imdb_id == '?i=tt0079501') assert (search.url == 'http://www.omdbapi.com/?i=tt0079501&tomatoes=true')
def test_series_title(): search = Search() search.title('Game+of+Thrones') assert(search.m_title == u'?t=Game+of+Thrones') assert(search.url == 'http://www.omdbapi.com/?t=Game+of+Thrones&y=&plot=short&tomatoes=true')
def test_movie_title(): search = Search() search.title('Mad+Max') assert(search.m_title == u'?t=Mad+Max') assert(search.url == 'http://www.omdbapi.com/?t=Mad+Max&y=&plot=short&tomatoes=true')
def __init__(self): self.search = Search() self.main_menu() self.movie_title = '' self.imdb_num = ''