mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 12:05:07 +00:00
imSp4rky
86f26c31d5
standard_resolution() replaces the aspect-ratio logic in naming and is shared by -q matching, export names and progress/hybrid labels, so 1920x1088, 1916x1076, portrait and DCI sizes get standard names.
84 lines
2.8 KiB
Python
84 lines
2.8 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", hdr_format: str = "SMPTE ST 2086", profile: str = ""
|
|
) -> 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",
|
|
hdr_format_commercial=commercial,
|
|
hdr_format=hdr_format,
|
|
hdr_format_profile=profile,
|
|
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") == ""
|
|
|
|
|
|
def test_a_profile_8_dv_track_with_an_unnamed_pq_base_gets_dv_hdr() -> None:
|
|
assert hdr_tag("Dolby Vision", Video.Range.DV, hdr_format="Dolby Vision", profile="dvhe.08") == "DV.HDR"
|
|
|
|
|
|
def test_a_profile_5_dv_track_stays_dv() -> None:
|
|
assert hdr_tag("Dolby Vision", Video.Range.DV, hdr_format="Dolby Vision", profile="dvhe.05") == "DV"
|
|
|
|
|
|
def test_a_profile_8_4_dv_track_with_an_hlg_base_stays_dv() -> None:
|
|
assert hdr_tag("Dolby Vision", Video.Range.DV, transfer="HLG", hdr_format="Dolby Vision", profile="dvhe.08") == "DV"
|
|
|
|
|
|
def test_a_dv_track_with_a_named_hdr10_base_keeps_dv_hdr() -> None:
|
|
assert hdr_tag("HDR10", Video.Range.DV, hdr_format="Dolby Vision", profile="dvhe.08") == "DV.HDR"
|