bazarr/tests/subliminal_patch/test_subdivx.py

187 lines
5.8 KiB
Python

# -*- coding: utf-8 -*-
import pytest
import copy
from subliminal_patch.providers.subdivx import SubdivxSubtitlesProvider
from subliminal_patch.providers.subdivx import SubdivxSubtitle
from subliminal_patch.core import SZProviderPool
from subliminal_patch.core import Episode
from subzero.language import Language
def test_list_subtitles_movie(movies):
item = movies["dune"]
with SubdivxSubtitlesProvider() as provider:
subtitles = provider.list_subtitles(item, {Language("spa", "MX")})
assert len(subtitles) >= 9
def test_list_subtitles_movie_with_year_fallback(movies):
item = list(movies.values())[0]
item.title = "Everything Everywhere All at Once"
item.year = 2022
with SubdivxSubtitlesProvider() as provider:
assert provider.list_subtitles(item, {Language("spa", "MX")})
def test_handle_multi_page_search(episodes):
with SubdivxSubtitlesProvider() as provider:
for _ in provider._handle_multi_page_search(
"Game Of Thrones", episodes["got_s03e10"]
):
pass
@pytest.mark.parametrize(
"episode_key,expected", [("breaking_bad_s01e01", 15), ("inexistent", 0)]
)
def test_list_subtitles_episode(episodes, episode_key, expected):
item = episodes[episode_key]
with SubdivxSubtitlesProvider() as provider:
subtitles = provider.list_subtitles(item, {Language("spa", "MX")})
assert len(subtitles) >= expected
def test_list_subtitles_episode_with_year(episodes):
item = list(episodes.values())[0]
item.series = "The Twilight Zone"
item.name = "The Twilight Zone"
item.year = 1959
item.season = 1
item.episode = 1
with SubdivxSubtitlesProvider() as provider:
assert provider.list_subtitles(item, {Language.fromietf("es")})
def test_list_subtitles_castillian_spanish(episodes):
item = episodes["better_call_saul_s06e04"]
with SubdivxSubtitlesProvider() as provider:
assert provider.list_subtitles(item, {Language.fromietf("es")})
def test_list_subtitles_episode_with_title_only_fallback(episodes):
item = list(episodes.values())[0]
item.series = "The Bear"
item.name = "The Bear"
item.season = 1
item.episode = 1
with SubdivxSubtitlesProvider() as provider:
subtitles = provider.list_subtitles(item, {Language("spa", "MX")})
assert len(subtitles) > 2
def test_list_subtitles_episode_with_episode_title_fallback(episodes):
item = list(episodes.values())[0]
item.series = "30 for 30"
item.title = "The Two Escobars"
item.season = 1
item.episode = 16
with SubdivxSubtitlesProvider() as provider:
sub = provider.list_subtitles(item, {Language("spa", "MX")})[0]
assert sub.get_matches(item)
provider.download_subtitle(sub)
assert sub.is_valid()
def test_download_subtitle(movies):
subtitle = SubdivxSubtitle(
Language("spa", "MX"),
movies["dune"],
"https://www.subdivx.com/X66XNjMxMTAxX-dune--2021-aka-dune-part-one.html",
"Dune",
"",
"",
"https://www.subdivx.com/bajar.php?id=631101&u=9",
)
with SubdivxSubtitlesProvider() as provider:
provider.download_subtitle(subtitle)
assert subtitle.content is not None
def test_download_subtitle_episode_pack(episodes):
video = copy.copy(episodes["breaking_bad_s01e01"])
video.episode = 3
subtitle = SubdivxSubtitle(
Language("spa", "MX"),
video,
"https://www.subdivx.com/X66XMzY1NjEwX-breaking-bad-s01e0107.html",
"Breaking Bad S01E01-07",
"Son los del torrent que vienen Formato / Dimensiones 624x352 / Tamaño 351 MB -Incluye los Torrents-",
"",
"https://www.subdivx.com/bajar.php?id=365610&u=7",
)
with SubdivxSubtitlesProvider() as provider:
provider.download_subtitle(subtitle)
assert subtitle.content is not None
@pytest.fixture
def video():
return Episode(
**{
"name": "/tv/SEAL Team/Season 5/SEAL.Team.S05E11.1080p.WEB.H264-CAKES[rarbg].mkv",
"release_group": "CAKES",
"series": "SEAL Team",
"source": "Web",
"resolution": "1080p",
"video_codec": "H.264",
"season": 5,
"episode": 11,
"title": "Violence of Action",
"alternative_series": [],
}
)
def test_subtitle_description_not_lowercase(video):
with SubdivxSubtitlesProvider() as provider:
subtitles = provider.list_subtitles(video, {Language("spa", "MX")})
assert subtitles
assert not subtitles[0]._description.islower()
def test_subtitle_matches(video):
subtitle = SubdivxSubtitle(
Language("spa", "MX"),
video,
"SEAL Team S05E10",
"https://www.subdivx.com/X66XNjM1MTAxX-seal-team-s05e10.html",
(
"Mi subtítulo y sincronización en Español neutro, para SEAL TEAM -Temp 05x10 "
"Head On para WEB.H265-GGWP, NTb 1080p, WEB-H264.CAKES, WEBRip.x264-ION10 y "
"otras seguramente, gracias por sus comentarios, saludos."
),
"tolobich",
"https://www.subdivx.com/bajar.php?id=635101&u=9",
)
matches = subtitle.get_matches(video)
assert "source" in matches
assert "resolution" in matches
assert "video_codec" in matches
assert "release_group" in matches
def test_latin_1_subtitles():
item = Episode.fromname(
"/tv/Grey's Anatomy/Season 19/Greys.Anatomy.S19E13.1080p.WEB.h264-ELEANOR[rarbg].mkv"
)
item.series = "Grey's Anatomy"
item.season = 19
item.episode = 13
with SubdivxSubtitlesProvider() as provider:
subtitles = provider.list_subtitles(item, {Language.fromietf("es")})
subtitle = subtitles[0]
provider.download_subtitle(subtitle)
assert subtitle.is_valid()