You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
bazarr/tests/subliminal_patch/test_core.py

169 lines
5.2 KiB

from pathlib import Path
import pytest
from subliminal_patch import core
def test_scan_video_movie(tmpdir):
video_path = Path(tmpdir, "Taxi Driver 1976 Bluray 720p x264.mkv")
video_path.touch()
result = core.scan_video(str(video_path))
assert isinstance(result, core.Movie)
def test_scan_video_episode(tmpdir):
video_path = Path(tmpdir, "The Wire S01E01 Bluray 720p x264.mkv")
video_path.touch()
result = core.scan_video(str(video_path))
assert isinstance(result, core.Episode)
@pytest.fixture
def pool_instance():
yield core.SZProviderPool({"argenteam"}, {})
def test_pool_update_w_nothing(pool_instance):
pool_instance.update({}, {}, [], {})
assert pool_instance.providers == set()
assert pool_instance.discarded_providers == set()
def test_pool_update_w_multiple_providers(pool_instance):
assert pool_instance.providers == {"argenteam"}
pool_instance.update({"argenteam", "subdivx", "subf2m"}, {}, [], {})
assert pool_instance.providers == {"argenteam", "subdivx", "subf2m"}
def test_pool_update_discarded_providers(pool_instance):
assert pool_instance.providers == {"argenteam"}
# Provider was discarded internally
pool_instance.discarded_providers = {"argenteam"}
assert pool_instance.discarded_providers == {"argenteam"}
# Provider is set to be used again
pool_instance.update({"subdivx", "argenteam"}, {}, [], {})
assert pool_instance.providers == {"argenteam", "subdivx"}
# Provider should disappear from discarded providers
assert pool_instance.discarded_providers == set()
def test_pool_update_discarded_providers_2(pool_instance):
assert pool_instance.providers == {"argenteam"}
# Provider was discarded internally
pool_instance.discarded_providers = {"argenteam"}
assert pool_instance.discarded_providers == {"argenteam"}
# Provider is not set to be used again
pool_instance.update({"subdivx"}, {}, [], {})
assert pool_instance.providers == {"subdivx"}
# Provider should not disappear from discarded providers
assert pool_instance.discarded_providers == {"argenteam"}
def test_language_equals_init():
assert core._LanguageEquals([(core.Language("spa"), core.Language("spa", "MX"))])
def test_language_equals_init_invalid():
with pytest.raises(ValueError):
assert core._LanguageEquals([(core.Language("spa", "MX"),)])
def test_language_equals_init_empty_list_gracefully():
assert core._LanguageEquals([]) == []
@pytest.mark.parametrize(
"langs",
[
[(core.Language("spa"), core.Language("spa", "MX"))],
[(core.Language("por"), core.Language("por", "BR"))],
[(core.Language("zho"), core.Language("zho", "TW"))],
],
)
def test_language_equals_check_set(langs):
equals = core._LanguageEquals(langs)
lang_set = {langs[0]}
assert equals.check_set(lang_set) == set(langs)
def test_language_equals_check_set_do_nothing():
equals = core._LanguageEquals([(core.Language("eng"), core.Language("spa"))])
lang_set = {core.Language("spa")}
assert equals.check_set(lang_set) == {core.Language("spa")}
def test_language_equals_check_set_do_nothing_w_forced():
equals = core._LanguageEquals(
[(core.Language("spa", forced=True), core.Language("spa", "MX"))]
)
lang_set = {core.Language("spa")}
assert equals.check_set(lang_set) == {core.Language("spa")}
@pytest.fixture
def language_equals_pool_intance():
equals = [(core.Language("spa"), core.Language("spa", "MX"))]
yield core.SZProviderPool({"subdivx"}, language_equals=equals)
def test_language_equals_pool_intance_list_subtitles(
language_equals_pool_intance, movies
):
subs = language_equals_pool_intance.list_subtitles(
movies["dune"], {core.Language("spa")}
)
assert subs
assert all(sub.language == core.Language("spa", "MX") for sub in subs)
def test_language_equals_pool_intance_list_subtitles_reversed(movies):
equals = [(core.Language("spa", "MX"), core.Language("spa"))]
language_equals_pool_intance = core.SZProviderPool(
{"subdivx"}, language_equals=equals
)
subs = language_equals_pool_intance.list_subtitles(
movies["dune"], {core.Language("spa")}
)
assert subs
assert all(sub.language == core.Language("spa") for sub in subs)
def test_language_equals_pool_intance_list_subtitles_empty_lang_equals(movies):
language_equals_pool_intance = core.SZProviderPool(
{"subdivx"}, language_equals=None
)
subs = language_equals_pool_intance.list_subtitles(
movies["dune"], {core.Language("spa")}
)
assert subs
assert not all(sub.language == core.Language("spa", "MX") for sub in subs)
def test_language_equals_pool_intance_list_subtitles_return_nothing(movies):
equals = [
(core.Language("spa", "MX"), core.Language("eng")),
(core.Language("spa"), core.Language("eng")),
]
language_equals_pool_intance = core.SZProviderPool(
{"subdivx"}, language_equals=equals
)
subs = language_equals_pool_intance.list_subtitles(
movies["dune"], {core.Language("spa")}
)
assert not language_equals_pool_intance.download_best_subtitles(
subs, movies["dune"], {core.Language("spa")}
)