def test_list_subtitles(episodes):
video = episodes['bbt_s07e05']
languages = {Language('eng'), Language('fra')}
expected_subtitles = {249592, 249499, 32596, 249518}
with TVsubtitlesProvider() as provider:
subtitles = provider.list_subtitles(video, languages)
assert {subtitle.subtitle_id for subtitle in subtitles} == expected_subtitles
assert {subtitle.language for subtitle in subtitles} == languages
def test_download_subtitle(episodes):
video = episodes['bbt_s07e05']
languages = {Language('eng'), Language('fra')}
with TVsubtitlesProvider() as provider:
subtitles = provider.list_subtitles(video, languages)
provider.download_subtitle(subtitles[0])
assert subtitles[0].content is not None
assert subtitles[0].is_valid() is True
def test_query_no_year(episodes):
video = episodes['dallas_s01e03']
expected_subtitles = {124753}
with TVsubtitlesProvider() as provider:
subtitles = provider.query(video.series, video.season, video.episode,
video.year)
assert {subtitle.subtitle_id
for subtitle in subtitles} == expected_subtitles
def test_list_subtitles_episode_alternative_series(episodes):
video = episodes['turn_s03e01']
languages = {Language('fra')}
expected_subtitles = {307588}
with TVsubtitlesProvider() as provider:
subtitles = provider.list_subtitles(video, languages)
assert {subtitle.subtitle_id
for subtitle in subtitles} == expected_subtitles
assert {subtitle.language for subtitle in subtitles} == languages
def test_query(episodes):
video = episodes['bbt_s07e05']
expected_subtitles = {
268673, 249733, 249518, 249519, 249714, 32596, 249590, 249592, 249499,
261214
}
with TVsubtitlesProvider() as provider:
subtitles = provider.query(video.series, video.season, video.episode,
video.year)
assert {subtitle.subtitle_id
for subtitle in subtitles} == expected_subtitles
def test_search_show_id_us():
with TVsubtitlesProvider() as provider:
show_id = provider.search_show_id('House of Cards', 2013)
assert show_id == 1246
def test_search_show_id_ambiguous():
with TVsubtitlesProvider() as provider:
show_id = provider.search_show_id('New Girl')
assert show_id == 977
def test_search_show_id_incomplete():
with TVsubtitlesProvider() as provider:
show_id = provider.search_show_id('The Big Bang')
assert show_id is None
def test_search_show_id():
with TVsubtitlesProvider() as provider:
show_id = provider.search_show_id('The Big Bang Theory')
assert show_id == 154
def test_query_wrong_episode(episodes):
video = episodes['bbt_s07e05']
with TVsubtitlesProvider() as provider:
subtitles = provider.query(video.series, video.season, 55, video.year)
assert len(subtitles) == 0
def test_get_episode_ids_wrong_season():
with TVsubtitlesProvider() as provider:
episode_ids = provider.get_episode_ids(154, 55)
assert len(episode_ids) == 0
def test_search_show_id_error():
with TVsubtitlesProvider() as provider:
show_id = provider.search_show_id('The Big How I Met Your Mother')
assert show_id is None
def test_search_show_id_year_in_title():
with TVsubtitlesProvider() as provider:
show_id = provider.search_show_id('Dallas', 2012)
assert show_id == 1127
def test_search_show_id_no_year():
with TVsubtitlesProvider() as provider:
show_id = provider.search_show_id('Dallas')
assert show_id == 646
def test_search_show_id_uk():
with TVsubtitlesProvider() as provider:
show_id = provider.search_show_id('Beautiful People')
assert show_id == 657