603 lines
22 KiB
Python
Executable File
603 lines
22 KiB
Python
Executable File
"""watermark_config 单元测试 - wave163
|
||
|
||
覆盖：
|
||
- WatermarkConfig 数据类 / from_dict / validate / has_effect
|
||
- calc_position 9宫格位置计算
|
||
- calc_scroll_x 滚动计算
|
||
- build_image_watermark_filter 图片水印滤镜
|
||
- build_text_watermark_filter 文字水印滤镜
|
||
- get_position_names / get_position_display_name
|
||
- 常量验证
|
||
"""
|
||
|
||
import pytest
|
||
|
||
from packages.domain.watermark_config import (
|
||
DEFAULT_FONT_COLOR,
|
||
DEFAULT_FONT_SIZE,
|
||
DEFAULT_MARGIN_X,
|
||
DEFAULT_MARGIN_Y,
|
||
DEFAULT_MODE,
|
||
DEFAULT_OPACITY,
|
||
DEFAULT_POSITION,
|
||
DEFAULT_SCALE,
|
||
DEFAULT_SCROLL_SPEED,
|
||
VALID_POSITIONS,
|
||
WATERMARK_POSITIONS,
|
||
WatermarkConfig,
|
||
build_image_watermark_filter,
|
||
build_text_watermark_filter,
|
||
calc_position,
|
||
calc_scroll_x,
|
||
get_position_display_name,
|
||
get_position_names,
|
||
)
|
||
|
||
# ============================================================
|
||
# 常量
|
||
# ============================================================
|
||
|
||
|
||
class TestConstants:
|
||
def test_nine_positions(self):
|
||
assert len(VALID_POSITIONS) == 9
|
||
|
||
def test_watermark_positions_contains_all(self):
|
||
assert set(WATERMARK_POSITIONS.keys()) == VALID_POSITIONS
|
||
|
||
def test_default_position_valid(self):
|
||
assert DEFAULT_POSITION in VALID_POSITIONS
|
||
|
||
def test_default_values(self):
|
||
assert DEFAULT_MODE in ("text", "image")
|
||
assert 0 < DEFAULT_SCALE <= 1.0
|
||
assert 0 <= DEFAULT_OPACITY <= 1.0
|
||
assert DEFAULT_FONT_SIZE > 0
|
||
assert DEFAULT_MARGIN_X >= 0
|
||
assert DEFAULT_MARGIN_Y >= 0
|
||
assert DEFAULT_SCROLL_SPEED > 0
|
||
assert isinstance(DEFAULT_FONT_COLOR, str)
|
||
|
||
|
||
# ============================================================
|
||
# WatermarkConfig 默认值
|
||
# ============================================================
|
||
|
||
|
||
class TestWatermarkConfigDefaults:
|
||
def test_default_constructor(self):
|
||
config = WatermarkConfig()
|
||
assert config.mode == DEFAULT_MODE
|
||
assert config.position == DEFAULT_POSITION
|
||
assert config.scale == DEFAULT_SCALE
|
||
assert config.opacity == DEFAULT_OPACITY
|
||
assert config.font_size == DEFAULT_FONT_SIZE
|
||
assert config.font_color == DEFAULT_FONT_COLOR
|
||
assert config.margin_x == DEFAULT_MARGIN_X
|
||
assert config.margin_y == DEFAULT_MARGIN_Y
|
||
assert config.scroll is False
|
||
assert config.scroll_speed == DEFAULT_SCROLL_SPEED
|
||
|
||
def test_text_mode_default(self):
|
||
config = WatermarkConfig()
|
||
assert config.mode == "text"
|
||
assert config.text == ""
|
||
assert config.font_path == ""
|
||
|
||
def test_image_mode_default(self):
|
||
config = WatermarkConfig(mode="image")
|
||
assert config.image_path == ""
|
||
|
||
|
||
# ============================================================
|
||
# from_dict
|
||
# ============================================================
|
||
|
||
|
||
class TestFromDict:
|
||
def test_none_returns_none(self):
|
||
assert WatermarkConfig.from_dict(None) is None
|
||
|
||
def test_empty_dict_returns_none(self):
|
||
assert WatermarkConfig.from_dict({}) is None
|
||
|
||
def test_disabled_returns_none(self):
|
||
assert WatermarkConfig.from_dict({"enabled": False}) is None
|
||
|
||
def test_image_mode_missing_path_returns_none(self):
|
||
result = WatermarkConfig.from_dict({"enabled": True, "mode": "image"})
|
||
assert result is None
|
||
|
||
def test_image_mode_with_image_field(self):
|
||
result = WatermarkConfig.from_dict({"enabled": True, "mode": "image", "image": "/path/to/wm.png"})
|
||
assert result is not None
|
||
assert result.mode == "image"
|
||
assert result.image_path == "/path/to/wm.png"
|
||
|
||
def test_image_mode_with_image_path_field(self):
|
||
result = WatermarkConfig.from_dict({"enabled": True, "mode": "image", "image_path": "/path/wm.png"})
|
||
assert result is not None
|
||
assert result.image_path == "/path/wm.png"
|
||
|
||
def test_text_mode_missing_text_returns_none(self):
|
||
result = WatermarkConfig.from_dict({"enabled": True, "mode": "text"})
|
||
assert result is None
|
||
|
||
def test_text_mode_with_text(self):
|
||
result = WatermarkConfig.from_dict({"enabled": True, "mode": "text", "text": "hello"})
|
||
assert result is not None
|
||
assert result.mode == "text"
|
||
assert result.text == "hello"
|
||
|
||
def test_invalid_position_falls_back_default(self):
|
||
result = WatermarkConfig.from_dict(
|
||
{
|
||
"enabled": True,
|
||
"mode": "text",
|
||
"text": "hello",
|
||
"position": "invalid_pos",
|
||
}
|
||
)
|
||
assert result is not None
|
||
assert result.position == DEFAULT_POSITION
|
||
|
||
def test_custom_values_propagated(self):
|
||
result = WatermarkConfig.from_dict(
|
||
{
|
||
"enabled": True,
|
||
"mode": "text",
|
||
"text": "测试",
|
||
"position": "top_left",
|
||
"opacity": 0.5,
|
||
"font_size": 32,
|
||
"font_color": "red",
|
||
"margin_x": 30,
|
||
"margin_y": 40,
|
||
"scroll": True,
|
||
"scroll_speed": 100,
|
||
}
|
||
)
|
||
assert result is not None
|
||
assert result.position == "top_left"
|
||
assert result.opacity == 0.5
|
||
assert result.font_size == 32
|
||
assert result.font_color == "red"
|
||
assert result.margin_x == 30
|
||
assert result.margin_y == 40
|
||
assert result.scroll is True
|
||
assert result.scroll_speed == 100
|
||
|
||
def test_image_mode_custom_values(self):
|
||
result = WatermarkConfig.from_dict(
|
||
{
|
||
"enabled": True,
|
||
"mode": "image",
|
||
"image_path": "/wm.png",
|
||
"scale": 0.3,
|
||
"opacity": 0.6,
|
||
}
|
||
)
|
||
assert result is not None
|
||
assert result.scale == 0.3
|
||
assert result.opacity == 0.6
|
||
|
||
def test_default_mode_when_unspecified(self):
|
||
# 只给enabled和text，mode默认text
|
||
result = WatermarkConfig.from_dict({"enabled": True, "text": "hello"})
|
||
assert result is not None
|
||
assert result.mode == DEFAULT_MODE
|
||
|
||
def test_text_empty_string_returns_none(self):
|
||
result = WatermarkConfig.from_dict({"enabled": True, "mode": "text", "text": ""})
|
||
assert result is None
|
||
|
||
|
||
# ============================================================
|
||
# validate
|
||
# ============================================================
|
||
|
||
|
||
class TestValidate:
|
||
def test_valid_text_config(self):
|
||
config = WatermarkConfig(mode="text", text="hello")
|
||
ok, msg = config.validate()
|
||
assert ok is True
|
||
assert msg == ""
|
||
|
||
def test_valid_image_config(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png")
|
||
ok, msg = config.validate()
|
||
assert ok is True
|
||
assert msg == ""
|
||
|
||
def test_invalid_position(self):
|
||
config = WatermarkConfig(mode="text", text="hi", position="nowhere")
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "位置" in msg
|
||
|
||
def test_opacity_too_high(self):
|
||
config = WatermarkConfig(mode="text", text="hi", opacity=1.5)
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "透明度" in msg
|
||
|
||
def test_opacity_negative(self):
|
||
config = WatermarkConfig(mode="text", text="hi", opacity=-0.1)
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "透明度" in msg
|
||
|
||
def test_opacity_boundary_zero(self):
|
||
config = WatermarkConfig(mode="text", text="hi", opacity=0.0)
|
||
ok, _ = config.validate()
|
||
assert ok is True
|
||
|
||
def test_opacity_boundary_one(self):
|
||
config = WatermarkConfig(mode="text", text="hi", opacity=1.0)
|
||
ok, _ = config.validate()
|
||
assert ok is True
|
||
|
||
def test_image_missing_path(self):
|
||
config = WatermarkConfig(mode="image")
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "图片路径" in msg
|
||
|
||
def test_image_scale_too_small(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", scale=0.001)
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "缩放" in msg
|
||
|
||
def test_image_scale_too_large(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", scale=1.5)
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "缩放" in msg
|
||
|
||
def test_image_scale_boundary_low(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", scale=0.01)
|
||
ok, _ = config.validate()
|
||
assert ok is True
|
||
|
||
def test_image_scale_boundary_high(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", scale=1.0)
|
||
ok, _ = config.validate()
|
||
assert ok is True
|
||
|
||
def test_text_empty(self):
|
||
config = WatermarkConfig(mode="text", text="")
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "文字内容" in msg
|
||
|
||
def test_font_size_zero(self):
|
||
config = WatermarkConfig(mode="text", text="hi", font_size=0)
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "字体大小" in msg
|
||
|
||
def test_font_size_negative(self):
|
||
config = WatermarkConfig(mode="text", text="hi", font_size=-5)
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "字体大小" in msg
|
||
|
||
def test_unknown_mode(self):
|
||
config = WatermarkConfig(mode="video", text="hi")
|
||
ok, msg = config.validate()
|
||
assert ok is False
|
||
assert "模式" in msg
|
||
|
||
|
||
# ============================================================
|
||
# has_effect
|
||
# ============================================================
|
||
|
||
|
||
class TestHasEffect:
|
||
def test_text_has_effect(self):
|
||
config = WatermarkConfig(mode="text", text="hello", opacity=0.5, font_size=24)
|
||
assert config.has_effect() is True
|
||
|
||
def test_text_empty_no_effect(self):
|
||
config = WatermarkConfig(mode="text", text="")
|
||
assert config.has_effect() is False
|
||
|
||
def test_text_zero_opacity_no_effect(self):
|
||
config = WatermarkConfig(mode="text", text="hi", opacity=0.0)
|
||
assert config.has_effect() is False
|
||
|
||
def test_text_zero_font_size_no_effect(self):
|
||
config = WatermarkConfig(mode="text", text="hi", font_size=0)
|
||
assert config.has_effect() is False
|
||
|
||
def test_image_has_effect(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", opacity=0.5)
|
||
assert config.has_effect() is True
|
||
|
||
def test_image_no_path_no_effect(self):
|
||
config = WatermarkConfig(mode="image")
|
||
assert config.has_effect() is False
|
||
|
||
def test_image_zero_opacity_no_effect(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", opacity=0.0)
|
||
assert config.has_effect() is False
|
||
|
||
def test_unknown_mode_no_effect(self):
|
||
config = WatermarkConfig(mode="invalid")
|
||
assert config.has_effect() is False
|
||
|
||
|
||
# ============================================================
|
||
# calc_position
|
||
# ============================================================
|
||
|
||
|
||
class TestCalcPosition:
|
||
OUT_W = 1920
|
||
OUT_H = 1080
|
||
WM_W = 200
|
||
WM_H = 50
|
||
MX = 20
|
||
MY = 20
|
||
|
||
def test_top_left(self):
|
||
x, y = calc_position("top_left", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == 20
|
||
assert y == 20
|
||
|
||
def test_top_center(self):
|
||
x, y = calc_position("top_center", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == (1920 - 200) // 2
|
||
assert y == 20
|
||
|
||
def test_top_right(self):
|
||
x, y = calc_position("top_right", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == 1920 - 200 - 20
|
||
assert y == 20
|
||
|
||
def test_center_left(self):
|
||
x, y = calc_position("center_left", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == 20
|
||
assert y == (1080 - 50) // 2
|
||
|
||
def test_center(self):
|
||
x, y = calc_position("center", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == (1920 - 200) // 2
|
||
assert y == (1080 - 50) // 2
|
||
|
||
def test_center_right(self):
|
||
x, y = calc_position("center_right", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == 1920 - 200 - 20
|
||
assert y == (1080 - 50) // 2
|
||
|
||
def test_bottom_left(self):
|
||
x, y = calc_position("bottom_left", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == 20
|
||
assert y == 1080 - 50 - 20
|
||
|
||
def test_bottom_center(self):
|
||
x, y = calc_position("bottom_center", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == (1920 - 200) // 2
|
||
assert y == 1080 - 50 - 20
|
||
|
||
def test_bottom_right(self):
|
||
x, y = calc_position("bottom_right", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == 1920 - 200 - 20
|
||
assert y == 1080 - 50 - 20
|
||
|
||
def test_invalid_position_defaults_bottom_right(self):
|
||
x, y = calc_position("invalid", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
assert x == 1920 - 200 - 20
|
||
assert y == 1080 - 50 - 20
|
||
|
||
def test_zero_margins(self):
|
||
x, y = calc_position("top_left", self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, 0, 0)
|
||
assert x == 0
|
||
assert y == 0
|
||
|
||
def test_all_nine_positions_are_unique(self):
|
||
positions = set()
|
||
for pos in VALID_POSITIONS:
|
||
pos_xy = calc_position(pos, self.OUT_W, self.OUT_H, self.WM_W, self.WM_H, self.MX, self.MY)
|
||
positions.add(pos_xy)
|
||
assert len(positions) == 9
|
||
|
||
|
||
# ============================================================
|
||
# calc_scroll_x
|
||
# ============================================================
|
||
|
||
|
||
class TestCalcScrollX:
|
||
def test_returns_string(self):
|
||
result = calc_scroll_x("bottom_left", 1920, 200, 50)
|
||
assert isinstance(result, str)
|
||
|
||
def test_contains_output_width(self):
|
||
result = calc_scroll_x("bottom_left", 1920, 200, 50)
|
||
assert "1920" in result
|
||
|
||
def test_contains_wm_width(self):
|
||
result = calc_scroll_x("bottom_left", 1920, 200, 50)
|
||
assert "200" in result
|
||
|
||
def test_contains_speed(self):
|
||
result = calc_scroll_x("bottom_left", 1920, 200, 50)
|
||
assert "50" in result
|
||
|
||
def test_contains_mod_keyword(self):
|
||
result = calc_scroll_x("bottom_left", 1920, 200, 50)
|
||
assert "mod" in result
|
||
|
||
|
||
# ============================================================
|
||
# build_image_watermark_filter
|
||
# ============================================================
|
||
|
||
|
||
class TestBuildImageWatermarkFilter:
|
||
def test_returns_tuple(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", position="top_left")
|
||
result = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
|
||
assert isinstance(result, tuple)
|
||
assert len(result) == 2
|
||
|
||
def test_filter_contains_overlay(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", position="bottom_right")
|
||
filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
|
||
assert "overlay" in filter_str
|
||
|
||
def test_filter_contains_scale(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", scale=0.3)
|
||
filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
|
||
assert "scale=" in filter_str
|
||
|
||
def test_opacity_applied_when_below_one(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", opacity=0.5)
|
||
filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
|
||
assert "colorchannelmixer" in filter_str
|
||
assert "aa=0.5" in filter_str
|
||
|
||
def test_opacity_one_no_mixer(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", opacity=1.0)
|
||
filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
|
||
assert "colorchannelmixer" not in filter_str
|
||
|
||
def test_input_args_contain_image_path(self):
|
||
config = WatermarkConfig(mode="image", image_path="/path/to/wm.png")
|
||
_, input_args = build_image_watermark_filter("[in]", "/path/to/wm.png", 1920, 1080, "[out]", config)
|
||
assert input_args == ["-i", "/path/to/wm.png"]
|
||
|
||
def test_output_label_present(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png")
|
||
filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[final]", config)
|
||
assert "[final]" in filter_str
|
||
|
||
def test_scroll_mode_contains_t_variable(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", scroll=True, scroll_speed=60)
|
||
filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
|
||
assert "*t" in filter_str or "t*" in filter_str
|
||
|
||
def test_static_mode_no_t_variable_in_overlay_x(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png", scroll=False)
|
||
filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
|
||
# 静态模式 x 是具体数值，不包含 t 变量
|
||
overlay_part = filter_str.split("overlay=")[1]
|
||
assert "t" not in overlay_part.split(":")[0] or "wm" in overlay_part.split(":")[0]
|
||
|
||
def test_semicolon_separated_filters(self):
|
||
config = WatermarkConfig(mode="image", image_path="/wm.png")
|
||
filter_str, _ = build_image_watermark_filter("[in]", "/wm.png", 1920, 1080, "[out]", config)
|
||
assert ";" in filter_str
|
||
|
||
|
||
# ============================================================
|
||
# build_text_watermark_filter
|
||
# ============================================================
|
||
|
||
|
||
class TestBuildTextWatermarkFilter:
|
||
def test_returns_string(self):
|
||
config = WatermarkConfig(mode="text", text="hello")
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
assert isinstance(result, str)
|
||
|
||
def test_contains_drawtext(self):
|
||
config = WatermarkConfig(mode="text", text="hello")
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
assert "drawtext=" in result
|
||
|
||
def test_contains_text(self):
|
||
config = WatermarkConfig(mode="text", text="测试水印")
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
assert "测试水印" in result
|
||
|
||
def test_fontsize_in_filter(self):
|
||
config = WatermarkConfig(mode="text", text="hi", font_size=36)
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
assert "fontsize=36" in result
|
||
|
||
def test_fontcolor_with_opacity(self):
|
||
config = WatermarkConfig(mode="text", text="hi", font_color="red", opacity=0.5)
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
assert "fontcolor=red@0.5" in result
|
||
|
||
def test_font_path_included(self):
|
||
config = WatermarkConfig(mode="text", text="hi", font_path="/fonts/simsun.ttf")
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
assert "fontfile=" in result
|
||
assert "simsun.ttf" in result
|
||
|
||
def test_no_font_path_when_empty(self):
|
||
config = WatermarkConfig(mode="text", text="hi", font_path="")
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
assert "fontfile=" not in result
|
||
|
||
def test_text_colon_escaped(self):
|
||
config = WatermarkConfig(mode="text", text="time: 00:00")
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
# 冒号应该被转义
|
||
assert "time\\:" in result or "time\\\\:" in result
|
||
|
||
def test_scroll_mode_contains_mod(self):
|
||
config = WatermarkConfig(mode="text", text="scrolling", scroll=True, scroll_speed=50)
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
assert "mod(" in result
|
||
|
||
def test_static_mode_has_numeric_x_y(self):
|
||
config = WatermarkConfig(mode="text", text="hi", position="top_left")
|
||
result = build_text_watermark_filter("[in]", "[out]", config, 1920, 1080)
|
||
assert "x=" in result
|
||
assert "y=" in result
|
||
|
||
def test_output_label_present(self):
|
||
config = WatermarkConfig(mode="text", text="hi")
|
||
result = build_text_watermark_filter("[in]", "[final_out]", config, 1920, 1080)
|
||
assert "[final_out]" in result
|
||
|
||
def test_input_label_present(self):
|
||
config = WatermarkConfig(mode="text", text="hi")
|
||
result = build_text_watermark_filter("[video_in]", "[out]", config, 1920, 1080)
|
||
assert result.startswith("[video_in]")
|
||
|
||
|
||
# ============================================================
|
||
# 工具函数
|
||
# ============================================================
|
||
|
||
|
||
class TestGetPositionNames:
|
||
def test_returns_nine_names(self):
|
||
names = get_position_names()
|
||
assert len(names) == 9
|
||
|
||
def test_order_is_correct(self):
|
||
names = get_position_names()
|
||
# 按从上到下、从左到右
|
||
assert names[0] == "top_left"
|
||
assert names[1] == "top_center"
|
||
assert names[2] == "top_right"
|
||
assert names[3] == "center_left"
|
||
assert names[4] == "center"
|
||
assert names[5] == "center_right"
|
||
assert names[6] == "bottom_left"
|
||
assert names[7] == "bottom_center"
|
||
assert names[8] == "bottom_right"
|
||
|
||
def test_all_valid(self):
|
||
names = get_position_names()
|
||
assert set(names) == VALID_POSITIONS
|
||
|
||
|
||
class TestGetPositionDisplayName:
|
||
def test_known_position(self):
|
||
assert get_position_display_name("top_left") == "左上"
|
||
assert get_position_display_name("center") == "中心"
|
||
assert get_position_display_name("bottom_right") == "右下"
|
||
|
||
def test_unknown_position_returns_original(self):
|
||
assert get_position_display_name("invalid") == "invalid"
|