bazarr/tests/subliminal_patch/conftest.py

160 lines
4.7 KiB
Python

# -*- coding: utf-8 -*-
import logging
import os
import pytest
from subliminal_patch.core import Movie, Episode
from subzero.language import Language
logging.getLogger("vcr").setLevel(logging.WARNING)
logging.getLogger("rebulk").setLevel(logging.WARNING)
@pytest.fixture(scope="module")
def vcr_cassette_dir(request):
return os.path.join("tests/subliminal_patch/cassettes", request.module.__name__)
@pytest.fixture
def movies():
return {
"dune": Movie(
"Dune.2021.1080p.WEBRip.DD5.1.x264-SHITBOX",
"Dune",
year=2021,
resolution="1080p",
source="Web",
# other="Rip",
imdb_id="tt1160419",
alternative_titles=["Dune: Part One"],
audio_codec="Dolby Digital",
video_codec="H.264",
release_group="SHITBOX",
),
"man_of_steel": Movie(
os.path.join(
"Man of Steel (2013)", "man.of.steel.2013.720p.bluray.x264-felony.mkv"
),
"Man of Steel",
source="Blu-Ray",
release_group="felony",
resolution="720p",
video_codec="H.264",
audio_codec="DTS",
imdb_id="tt0770828",
size=7033732714,
year=2013,
hashes={
"napiprojekt": "6303e7ee6a835e9fcede9fb2fb00cb36",
"opensubtitles": "5b8f8f4e41ccb21e",
"shooter": "314f454ab464775498ae6f1f5ad813a9;fdaa8b702d8936feba2122e93ba5c44f;"
"0a6935e3436aa7db5597ef67a2c494e3;4d269733f36ddd49f71e92732a462fe5",
"thesubdb": "ad32876133355929d814457537e12dc2",
},
),
"enders_game": Movie(
"enders.game.2013.720p.bluray.x264-sparks.mkv",
"Ender's Game",
source="Blu-Ray",
release_group="sparks",
resolution="720p",
video_codec="H.264",
year=2013,
),
"inexistent": Movie(
"inexistent.movie.2013.720p.bluray.x264-sparks.mkv",
"231231asdklsad2321",
source="Blu-Ray",
resolution="720p",
video_codec="H.264",
year=2050,
),
"blade_runner": Movie(
"Alien (1979) Theatrical HDR 1080p UHD BluRay x265 HEVC EAC3-SARTRE",
"Alien",
source="Ultra HD Blu-ray",
release_group="SARTRE",
resolution="1080p",
video_codec="H.265",
audio_codec="Dolby Digital Plus",
imdb_id="tt0078748",
year=1979,
),
}
@pytest.fixture
def episodes():
return {
"got_s03e10": Episode(
os.path.join(
"Game of Thrones",
"Season 03",
"Game.of.Thrones.S03E10.Mhysa.720p.WEB-DL.DD5.1.H.264-NTb.mkv",
),
"Game of Thrones",
3,
10,
title="Mhysa",
tvdb_id=4517466,
series_tvdb_id=121361,
series_imdb_id="tt0944947",
source="Web",
release_group="NTb",
resolution="720p",
video_codec="H.264",
audio_codec="AC3",
imdb_id="tt2178796",
size=2142810931,
hashes={
"napiprojekt": "6303e7ee6a835e9fcede9fb2fb00cb36",
"opensubtitles": "b850baa096976c22",
"shooter": "b02d992c04ad74b31c252bd5a097a036;ef1b32f873b2acf8f166fc266bdf011a;"
"82ce34a3bcee0c66ed3b26d900d31cca;78113770551f3efd1e2d4ec45898c59c",
"thesubdb": "b1f899c77f4c960b84b8dbf840d4e42d",
},
),
"breaking_bad_s01e01": Episode(
"Breaking.Bad.S01E01.720p.BluRay.X264-REWARD.mkv",
"Breaking Bad",
1,
1,
source="Blu-Ray",
release_group="REWARD",
resolution="720p",
video_codec="H.264",
),
"better_call_saul_s06e04": Episode(
"Better.Call.Saul.S06E04.720p.WEBRIP.X264.mkv",
"Better Call Saul",
6,
4,
source="Web",
resolution="720p",
video_codec="H.264",
),
"inexistent": Episode(
"Inexistent.TVShow.S01E01.720p.BluRay.X264-REWARD.mkv",
"121361asdfgh",
1,
1,
source="Blu-Ray",
release_group="REWARD",
resolution="720p",
video_codec="H.264",
),
}
@pytest.fixture
def languages():
return {"en": Language.fromietf("en"), "es-MX": Language("spa", "MX")}
@pytest.fixture
def data():
return os.path.join(os.path.abspath(os.path.dirname(__file__)), "data")