def test_list_subtitles(episodes):
video = episodes['bbt_s07e05']
languages = {Language('eng'), Language('fra')}
with TheSubDBProvider() as provider:
subtitles = provider.list_subtitles(video, languages)
assert len(subtitles) == 2
assert {subtitle.language for subtitle in subtitles} == languages
def test_query(movies):
video = movies['man_of_steel']
expected_languages = {Language('eng'), Language('por', 'BR')}
with TheSubDBProvider() as provider:
subtitles = provider.query(video.hashes['thesubdb'])
assert len(subtitles) == 2
assert {subtitle.language for subtitle in subtitles} == expected_languages
def test_download_subtitle(episodes):
video = episodes['bbt_s07e05']
languages = {Language('eng'), Language('fra')}
with TheSubDBProvider() 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_wrong_hash():
with TheSubDBProvider() as provider:
subtitles = provider.query('11223344556677899877665544332211')
assert len(subtitles) == 0