def verify_name_gen(curtest, tvdb_instance):
ep = EpisodeInfo(
seriesname=curtest["parsedseriesname"],
seasonnumber=curtest["seasonnumber"],
episodenumbers=curtest["episodenumbers"],
)
correctedSeriesName, epName = getEpisodeName(tvdb_instance, ep)
assert correctedSeriesName is not None, "Corrected series name was none"
assert epName is not None, "Episode name was None"
assertEquals(epName, curtest["episodenames"])
assertEquals(correctedSeriesName, curtest["correctedseriesname"])
def verify_name_gen(curtest, tvdb_instance):
if "seasonnumber" in curtest:
ep = EpisodeInfo(
seriesname = curtest['parsedseriesname'],
seasonnumber = curtest['seasonnumber'],
episodenumbers = curtest['episodenumbers'])
elif any([isinstance(x, datetime.date) for x in curtest['episodenumbers']]):
ep = DatedEpisodeInfo(
seriesname = curtest['parsedseriesname'],
episodenumbers = curtest['episodenumbers'])
else:
ep = NoSeasonEpisodeInfo(
seriesname = curtest['parsedseriesname'],
episodenumbers = curtest['episodenumbers'])
correctedSeriesName, epName = getEpisodeName(tvdb_instance, ep)
assert correctedSeriesName is not None, "Corrected series name was none"
assert epName is not None, "Episode name was None"
assertEquals(epName, curtest['episodenames'])
assertEquals(correctedSeriesName, curtest['correctedseriesname'])
