def test_mustgivestr(self): with self.assertRaises(TypeError): imdb.getseasons([]) with self.assertRaises(TypeError): imdb.getnumseasons([]) with self.assertRaises(TypeError): imdb.getepisodes([]) with self.assertRaises(TypeError): imdb.getnumepisodes([]) with self.assertRaises(TypeError): imdb.getshow([])
def test_seasons(self): seasons = imdb.getseasons(self.showname) self.assertGreaterEqual(len(seasons), 9)