def test_init(self, number, should_raise): if should_raise: with pytest.raises(Exception) as e: LostFilmSeason(number) message = six.text_type(e.value) assert 'int' in message assert 'tuple[int, int]' in six.text_type(e.value) assert 'SpecialSeason' in six.text_type(e.value) else: LostFilmSeason(number)
def test_add_episode_success(self): episode1 = LostFilmEpisode(1, 1) episode2 = LostFilmEpisode(1, 2) season = LostFilmSeason(1) season.add_episode(episode1) season.add_episode(episode2) assert len(season) == 2 assert season[1] == episode1 assert season[2] == episode2 assert list(season) == [episode1, episode2] assert list(reversed(season)) == [episode2, episode1]
def test_add_season_success(self): season1 = LostFilmSeason(1) season2 = LostFilmSeason(2) show = LostFilmShow('Show', u'Шоу', 'Show', 2017) assert show.last_season is None assert show.seasons_url == 'https://www.lostfilm.tv/series/Show/seasons' show.add_season(season1) show.add_season(season2) assert len(show) == 2 assert show[1] == season1 assert show[2] == season2 assert show.last_season == season2 assert list(show) == [season1, season2] assert list(reversed(show)) == [season2, season1]
def test_special_season_shouldnot_became_last_season(self): season1 = LostFilmSeason(SpecialSeasons.Additional) show = LostFilmShow('Show', u'Шоу', 'Show', 2017) assert show.last_season is None assert show.seasons_url == 'https://www.lostfilm.tv/series/Show/seasons' show.add_season(season1) assert show.last_season is None assert len(show) == 1 assert show[SpecialSeasons.Additional] == season1 assert list(show) == [season1] assert list(reversed(show)) == [season1]
def test_add_episode_failed(self): season1 = LostFilmSeason(2) show = LostFilmShow('Show', u'Шоу', 'Show', 2017) assert show.seasons_url == 'https://www.lostfilm.tv/series/Show/seasons' show.add_season(season1) with pytest.raises(Exception) as e: show.add_season(season1) message = six.text_type(e.value) assert six.text_type(season1.number) in message assert 'already' in message assert len(show) == 1 assert show[2] == season1 assert show.last_season == season1 assert list(show) == [season1] assert list(reversed(show)) == [season1]
def test_add_episode_failed(self): episode1 = LostFilmEpisode(2, 1) season = LostFilmSeason(2) season.add_episode(episode1) with pytest.raises(Exception) as e: season.add_episode(episode1) message = six.text_type(e.value) assert six.text_type(episode1.number) in message assert six.text_type(season.number) in message assert 'already' in message assert len(season) == 1 assert season[1] == episode1 assert list(season) == [episode1] assert list(reversed(season)) == [episode1]
def test_is_special_season_success(self, number, result): episode = LostFilmSeason(number) assert episode.is_special_season() == result