bazarr/tests/subliminal_patch/test_score.py

from subliminal_patch import score
from subliminal_patch.providers.karagarga import KaragargaSubtitle


# def __call__(self, matches, subtitle, video, hearing_impaired=None):


def test_compute_score_set_var(movies, languages):
    subtitle = KaragargaSubtitle(languages["en"], "", "", "")
    score.compute_score({"hash"}, subtitle, movies["dune"])


def test_compute_score_set_var_w_episode(episodes, languages):
    subtitle = KaragargaSubtitle(languages["en"], "", "", "")
    score.compute_score({"hash"}, subtitle, episodes["breaking_bad_s01e01"])


def test_compute_score_defaults():
    assert score.ComputeScore()._scores == score.DEFAULT_SCORES


def test_compute_score_custom_invalid():
    assert (
        score.ComputeScore({"movie": {"hash": 120}, "episode": {"hash": 321}})._scores
        == score.DEFAULT_SCORES
    )


def test_compute_score_custom_valid():
    scores_copy = score.DEFAULT_SCORES.copy()
    scores_copy["movie"]["release_group"] = 12
    scores_copy["movie"]["source"] = 8

    scores_ = score.ComputeScore(scores_copy)
    assert scores_._scores["movie"]["release_group"] == 12
    assert scores_._scores["movie"]["source"] == 8