为所有函数和类添加中文注释文档字符串
This commit is contained in:
@@ -3,26 +3,32 @@ from __future__ import annotations
|
||||
from app.exercises.dead_bug.rules import detect_diagonal_extension, has_required_visibility, is_ready_position
|
||||
from app.exercises.dead_bug.types import DeadBugMetrics, Point
|
||||
|
||||
|
||||
class TestDeadBugRules:
|
||||
"""死虫式规则函数单元测试"""
|
||||
|
||||
def _make_landmark(self, x=0.5, y=0.5, z=0.0, visibility=1.0):
|
||||
"""创建测试用Point对象"""
|
||||
return Point(x, y, z, visibility)
|
||||
|
||||
def _make_visible_landmarks(self):
|
||||
"""创建33个全可见的测试用关键点"""
|
||||
return [self._make_landmark() for _ in range(33)]
|
||||
|
||||
def test_has_required_visibility_all_visible(self):
|
||||
"""测试:所有关键点可见时应返回True"""
|
||||
lm = self._make_visible_landmarks()
|
||||
indices = (11, 12, 13, 14, 15, 16, 23, 24, 25, 26, 27, 28)
|
||||
assert has_required_visibility(lm, indices, 0.45)
|
||||
|
||||
def test_has_required_visibility_low(self):
|
||||
"""测试:关键点可见度过低时应返回False"""
|
||||
lm = self._make_visible_landmarks()
|
||||
lm[11] = self._make_landmark(visibility=0.1)
|
||||
indices = (11, 12, 13, 14, 15, 16, 23, 24, 25, 26, 27, 28)
|
||||
assert not has_required_visibility(lm, indices, 0.45)
|
||||
|
||||
def test_detect_diagonal_extension_none(self):
|
||||
"""测试:四肢均未伸展时应返回None"""
|
||||
metrics = DeadBugMetrics(
|
||||
left_arm_extended=False, right_arm_extended=False,
|
||||
left_leg_extended=False, right_leg_extended=False,
|
||||
@@ -33,6 +39,7 @@ class TestDeadBugRules:
|
||||
assert detect_diagonal_extension(metrics) is None
|
||||
|
||||
def test_detect_diagonal_extension_left_arm_right_leg(self):
|
||||
"""测试:左臂+右腿对角伸展检测"""
|
||||
metrics = DeadBugMetrics(
|
||||
left_arm_extended=True, right_arm_extended=False,
|
||||
left_leg_extended=False, right_leg_extended=True,
|
||||
@@ -43,6 +50,7 @@ class TestDeadBugRules:
|
||||
assert detect_diagonal_extension(metrics) == "left_arm_right_leg"
|
||||
|
||||
def test_is_ready_position(self):
|
||||
"""测试:膝盖弯曲且四肢收缩应识别为准备姿态"""
|
||||
metrics = DeadBugMetrics(
|
||||
left_arm_extended=False, right_arm_extended=False,
|
||||
left_leg_extended=False, right_leg_extended=False,
|
||||
@@ -53,6 +61,7 @@ class TestDeadBugRules:
|
||||
assert is_ready_position(metrics)
|
||||
|
||||
def test_is_not_ready_legs_extended(self):
|
||||
"""测试:腿部伸展时不识别为准备姿态"""
|
||||
metrics = DeadBugMetrics(
|
||||
left_arm_extended=False, right_arm_extended=False,
|
||||
left_leg_extended=True, right_leg_extended=False,
|
||||
|
||||
Reference in New Issue
Block a user