mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
"""Tests for the HDR tag in the filename context when MediaInfo misses HDR10+ or the HDR format."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import types
|
|
|
|
import pytest
|
|
|
|
from unshackle.core.config import config
|
|
from unshackle.core.titles.movie import Movie
|
|
from unshackle.core.tracks import Tracks, Video
|
|
|
|
|
|
class DummyService:
|
|
pass
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clean_config(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(config, "tag_rules", [])
|
|
monkeypatch.setattr(config, "language_tags", {})
|
|
monkeypatch.setattr(config, "codec_names", {})
|
|
|
|
|
|
def hdr_tag(commercial: str, range_: Video.Range, transfer: str = "PQ") -> str:
|
|
movie = Movie(id_="movie-0001", service=DummyService, name="The Film", year=2024, language="en")
|
|
movie.tracks = Tracks(Video(url="https://example.test/v.mpd", language="en", codec=Video.Codec.AV1, range_=range_))
|
|
video = types.SimpleNamespace(
|
|
width=1920,
|
|
height=1080,
|
|
format="AV1",
|
|
scan_type="Progressive",
|
|
other_display_aspect_ratio=["16:9"],
|
|
hdr_format_commercial=commercial,
|
|
hdr_format="SMPTE ST 2086",
|
|
transfer_characteristics=transfer,
|
|
transfer_characteristics_original=None,
|
|
frame_rate="23.976",
|
|
)
|
|
media_info = types.SimpleNamespace(video_tracks=[video], audio_tracks=[])
|
|
return str(movie.build_base_template_context(media_info)["hdr"])
|
|
|
|
|
|
def test_an_hdr10p_track_upgrades_a_plain_hdr10_tag() -> None:
|
|
assert hdr_tag("HDR10", Video.Range.HDR10P) == "HDR10P"
|
|
|
|
|
|
def test_an_hdr10_track_keeps_the_hdr_tag() -> None:
|
|
assert hdr_tag("HDR10", Video.Range.HDR10) == "HDR"
|
|
|
|
|
|
def test_mediainfo_hdr10p_is_unchanged() -> None:
|
|
assert hdr_tag("HDR10 / HDR10+", Video.Range.HDR10) == "HDR10P"
|
|
|
|
|
|
def test_pq_without_an_hdr_format_gets_the_hdr_tag() -> None:
|
|
assert hdr_tag("", Video.Range.HDR10) == "HDR"
|
|
|
|
|
|
def test_hlg_without_an_hdr_format_keeps_the_hlg_tag() -> None:
|
|
assert hdr_tag("", Video.Range.HLG, transfer="HLG") == "HLG"
|
|
|
|
|
|
def test_sdr_gets_no_hdr_tag() -> None:
|
|
assert hdr_tag("", Video.Range.SDR, transfer="BT.709") == ""
|