"""subtitle_style 领域模型单测 — 纯逻辑.""" from __future__ import annotations import pytest from packages.domain.subtitle_style import ( ALLOWED_SUBTITLE_EXTENSIONS, DEFAULT_COLOR, DEFAULT_FONT, DEFAULT_FONT_SIZE, DEFAULT_MAX_CHARS_PER_LINE, DEFAULT_POSITION, DEFAULT_STROKE_COLOR, DEFAULT_STROKE_WIDTH, POSITION_ALIASES, POSITION_ALIGNMENT, SubtitleSegment, SubtitleStyle, escape_ass_text, format_ass_time, hex_to_ass_bgr, hex_to_ass_color, opacity_to_ass_alpha, wrap_text, ) # ── 常量测试 ────────────────────────────────────────────────────────────────── class TestConstants: def test_position_alignment_has_9_positions(self): assert len(POSITION_ALIGNMENT) == 9 assert POSITION_ALIGNMENT["bottom_center"] == 2 assert POSITION_ALIGNMENT["top_center"] == 8 assert POSITION_ALIGNMENT["center"] == 5 def test_position_aliases(self): assert POSITION_ALIASES["top"] == "top_center" assert POSITION_ALIASES["bottom"] == "bottom_center" assert POSITION_ALIASES["middle"] == "center" def test_default_values(self): assert DEFAULT_FONT == "思源黑体" assert DEFAULT_FONT_SIZE == 24 assert DEFAULT_COLOR == "#FFFFFF" assert DEFAULT_POSITION == "bottom_center" def test_allowed_extensions(self): assert ".srt" in ALLOWED_SUBTITLE_EXTENSIONS assert ".ass" in ALLOWED_SUBTITLE_EXTENSIONS assert ".vtt" in ALLOWED_SUBTITLE_EXTENSIONS # ── 工具函数测试 ────────────────────────────────────────────────────────────── class TestHexToAssColor: def test_white(self): assert hex_to_ass_color("#FFFFFF") == "&H00FFFFFF" def test_black(self): assert hex_to_ass_color("#000000") == "&H00000000" def test_red(self): assert hex_to_ass_color("#FF0000") == "&H000000FF" def test_blue(self): assert hex_to_ass_color("#0000FF") == "&H00FF0000" def test_green(self): assert hex_to_ass_color("#00FF00") == "&H0000FF00" def test_without_hash(self): assert hex_to_ass_color("FF0000") == "&H000000FF" def test_lowercase(self): assert hex_to_ass_color("#ff0000") == "&H000000FF" def test_invalid_length_returns_white(self): assert hex_to_ass_color("#FFF") == "&H00FFFFFF" assert hex_to_ass_color("#FF000000") == "&H00FFFFFF" def test_empty_string(self): assert hex_to_ass_color("") == "&H00FFFFFF" class TestHexToAssBgr: def test_white(self): assert hex_to_ass_bgr("#FFFFFF") == "FFFFFF" def test_red(self): assert hex_to_ass_bgr("#FF0000") == "0000FF" def test_blue(self): assert hex_to_ass_bgr("#0000FF") == "FF0000" def test_invalid_length(self): assert hex_to_ass_bgr("#FFF") == "FFFFFF" class TestOpacityToAssAlpha: def test_fully_opaque(self): assert opacity_to_ass_alpha(1.0) == "00" def test_fully_transparent(self): assert opacity_to_ass_alpha(0.0) == "FF" def test_half(self): assert opacity_to_ass_alpha(0.5) == "80" def test_above_1_clamped(self): assert opacity_to_ass_alpha(1.5) == "00" def test_below_0_clamped(self): assert opacity_to_ass_alpha(-0.5) == "FF" class TestEscapeAssText: def test_newline_unix(self): assert escape_ass_text("hello\nworld") == "hello\\Nworld" def test_newline_windows(self): assert escape_ass_text("hello\r\nworld") == "hello\\Nworld" def test_newline_mac(self): assert escape_ass_text("hello\rworld") == "hello\\Nworld" def test_curly_braces(self): assert escape_ass_text("{text}") == "(text)" def test_mixed(self): assert escape_ass_text("hello\n{world}\r\nend") == "hello\\N(world)\\Nend" def test_empty(self): assert escape_ass_text("") == "" class TestFormatAssTime: def test_zero(self): assert format_ass_time(0.0) == "0:00:00.00" def test_seconds_only(self): assert format_ass_time(5.5) == "0:00:05.50" def test_minutes(self): assert format_ass_time(65.25) == "0:01:05.25" def test_hours(self): assert format_ass_time(3661.5) == "1:01:01.50" def test_negative_returns_zero(self): assert format_ass_time(-1.0) == "0:00:00.00" def test_centiseconds_precision(self): assert format_ass_time(1.234) == "0:00:01.23" class TestWrapText: def test_short_text_no_wrap(self): assert wrap_text("你好", 10) == ["你好"] def test_exact_length_no_wrap(self): text = "你" * 10 result = wrap_text(text, 10) assert len(result) == 1 assert len(result[0]) == 10 def test_long_text_breaks_at_max(self): text = "你" * 25 result = wrap_text(text, 10) assert len(result) == 3 assert len(result[0]) == 10 assert len(result[1]) == 10 assert len(result[2]) == 5 def test_breaks_at_punctuation(self): # "一二三四五六。七八九十"共10字，"。"在索引6 # max_chars=8 时，从 8 往回找到 4，会命中索引6的"。" text = "一二三四五六。七八九十" result = wrap_text(text, 8) assert len(result) == 2 assert result[0] == "一二三四五六。" assert result[1] == "七八九十" def test_no_punctuation_breaks_at_max(self): text = "一二三四五六七八九十一二三四五六七八九十" result = wrap_text(text, 10) assert len(result[0]) == 10 def test_empty_text(self): assert wrap_text("", 10) == [""] def test_zero_max_chars(self): assert wrap_text("hello", 0) == ["hello"] def test_negative_max_chars(self): result = wrap_text("hello", -5) assert isinstance(result, list) assert len(result) == 1 # ── SubtitleStyle 测试 ─────────────────────────────────────────────────────── class TestSubtitleStyleDefaults: def test_default_values(self): style = SubtitleStyle() assert style.font_name == DEFAULT_FONT assert style.font_size == DEFAULT_FONT_SIZE assert style.font_color == DEFAULT_COLOR assert style.bold is False assert style.italic is False assert style.stroke_enabled is True assert style.stroke_color == DEFAULT_STROKE_COLOR assert style.stroke_width == DEFAULT_STROKE_WIDTH assert style.position == DEFAULT_POSITION assert style.max_chars_per_line == DEFAULT_MAX_CHARS_PER_LINE class TestSubtitleStyleFromDict: def test_none_returns_default(self): style = SubtitleStyle.from_dict(None) assert style.font_name == DEFAULT_FONT def test_empty_dict_returns_default(self): style = SubtitleStyle.from_dict({}) assert style.font_size == DEFAULT_FONT_SIZE def test_custom_font(self): style = SubtitleStyle.from_dict({"font": "微软雅黑", "size": 32}) assert style.font_name == "微软雅黑" assert style.font_size == 32 def test_color(self): style = SubtitleStyle.from_dict({"color": "#FF0000"}) assert style.font_color == "#FF0000" def test_bold_italic(self): style = SubtitleStyle.from_dict({"bold": True, "italic": True}) assert style.bold is True assert style.italic is True def test_stroke_config(self): style = SubtitleStyle.from_dict( { "stroke_enabled": False, "stroke_color": "#00FF00", "stroke_width": 2.0, } ) assert style.stroke_enabled is False assert style.stroke_color == "#00FF00" assert style.stroke_width == 2.0 def test_shadow_config(self): style = SubtitleStyle.from_dict( { "shadow_enabled": True, "shadow_color": "#111111", "shadow_offset_x": 4, "shadow_offset_y": 4, "shadow_blur": 1.5, } ) assert style.shadow_enabled is True assert style.shadow_color == "#111111" assert style.shadow_offset_x == 4 assert style.shadow_offset_y == 4 assert style.shadow_blur == 1.5 def test_background_config(self): style = SubtitleStyle.from_dict( { "background_enabled": True, "background_color": "#000000", "background_opacity": 0.7, "background_padding": 10, "background_radius": 6, } ) assert style.background_enabled is True assert style.background_opacity == 0.7 assert style.background_padding == 10 def test_background_opacity_clamped_0_to_1(self): style = SubtitleStyle.from_dict({"background_opacity": -0.5}) assert style.background_opacity == 0.0 style2 = SubtitleStyle.from_dict({"background_opacity": 1.5}) assert style2.background_opacity == 1.0 def test_position_valid(self): style = SubtitleStyle.from_dict({"position": "top_center"}) assert style.position == "top_center" def test_position_alias(self): style = SubtitleStyle.from_dict({"position": "top"}) assert style.position == "top_center" def test_position_invalid_falls_back(self): style = SubtitleStyle.from_dict({"position": "invalid_pos"}) assert style.position == DEFAULT_POSITION def test_margins(self): style = SubtitleStyle.from_dict({"margin_v": 80, "margin_l": 50, "margin_r": 50}) assert style.margin_v == 80 assert style.margin_l == 50 assert style.margin_r == 50 def test_max_chars_per_line(self): style = SubtitleStyle.from_dict({"max_chars_per_line": 15}) assert style.max_chars_per_line == 15 def test_line_spacing(self): style = SubtitleStyle.from_dict({"line_spacing": 4}) assert style.line_spacing == 4 def test_fade_in_out(self): style = SubtitleStyle.from_dict({"fade_in": 0.5, "fade_out": 1.0}) assert style.fade_in == 0.5 assert style.fade_out == 1.0 def test_fade_negative_clamped(self): style = SubtitleStyle.from_dict({"fade_in": -1, "fade_out": -2}) assert style.fade_in == 0.0 assert style.fade_out == 0.0 def test_animation_type(self): style = SubtitleStyle.from_dict({"animation_type": "fade"}) assert style.animation_type == "fade" def test_invalid_int_falls_back(self): style = SubtitleStyle.from_dict({"size": "not_a_number"}) assert style.font_size == DEFAULT_FONT_SIZE def test_invalid_float_falls_back(self): style = SubtitleStyle.from_dict({"stroke_width": "abc"}) assert style.stroke_width == DEFAULT_STROKE_WIDTH class TestSubtitleStyleProperties: def test_alignment_bottom_center(self): style = SubtitleStyle(position="bottom_center") assert style.alignment == 2 def test_alignment_top_center(self): style = SubtitleStyle(position="top_center") assert style.alignment == 8 def test_ass_font_color(self): style = SubtitleStyle(font_color="#FF0000") assert style.ass_font_color == "&H000000FF" def test_ass_stroke_color(self): style = SubtitleStyle(stroke_color="#00FF00") assert style.ass_stroke_color == "&H0000FF00" def test_ass_shadow_color(self): style = SubtitleStyle(shadow_color="#0000FF") assert style.ass_shadow_color == "&H00FF0000" def test_ass_background_color(self): style = SubtitleStyle(background_color="#FF0000", background_opacity=0.5) # alpha = 255 - 127 = 128 = 0x80, bgr of red = 0000FF assert style.ass_background_color == "&H800000FF" # ── SubtitleSegment 测试 ───────────────────────────────────────────────────── class TestSubtitleSegment: def test_basic(self): seg = SubtitleSegment(start=1.0, end=3.0, text="你好") assert seg.start == 1.0 assert seg.end == 3.0 assert seg.text == "你好" assert seg.style_name == "Default" def test_custom_style(self): seg = SubtitleSegment(start=0, end=2, text="hi", style_name="Title") assert seg.style_name == "Title" def test_duration(self): seg = SubtitleSegment(start=1.5, end=4.0, text="test") assert seg.duration == 2.5 def test_duration_zero_when_end_before_start(self): seg = SubtitleSegment(start=5.0, end=3.0, text="test") assert seg.duration == 0.0 def test_is_valid_true(self): seg = SubtitleSegment(start=0, end=2, text="hello") assert seg.is_valid is True def test_is_valid_empty_text(self): seg = SubtitleSegment(start=0, end=2, text="") assert seg.is_valid is False def test_is_valid_zero_duration(self): seg = SubtitleSegment(start=1, end=1, text="hello") assert seg.is_valid is False