From c412698cff58a3073033bc045ffa68830b3eca8d Mon Sep 17 00:00:00 2001 From: iomgaa Date: Wed, 15 Jul 2026 06:30:14 -0400 Subject: [PATCH] =?UTF-8?q?style(test):=20=E4=BF=AE=20loader=20unit-sampli?= =?UTF-8?q?ng=20=E6=B5=8B=E8=AF=95=E7=9A=84=20ruff=20=E5=91=8A=E8=AD=A6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit TC003 将 Path 导入移入 TYPE_CHECKING 块(仅注解使用), C408 将 base = dict(...) 改为字典字面量。行为不变。 --- tests/unit/test_loader_unit_sampling.py | 29 ++++++++++++++----------- 1 file changed, 16 insertions(+), 13 deletions(-) diff --git a/tests/unit/test_loader_unit_sampling.py b/tests/unit/test_loader_unit_sampling.py index c8dc738..a91f343 100644 --- a/tests/unit/test_loader_unit_sampling.py +++ b/tests/unit/test_loader_unit_sampling.py @@ -13,7 +13,7 @@ from __future__ import annotations import json import random -from pathlib import Path +from typing import TYPE_CHECKING import pytest @@ -21,6 +21,9 @@ from app.harness.question_units import build_units from app.question_gen.loader import load_benchmark, stratified_sample from core.types import GeneratedQuestion +if TYPE_CHECKING: + from pathlib import Path + def _single(qid: str, task_type: str = "Single") -> GeneratedQuestion: """构造一道 single 题(无 pair 归属)。""" @@ -38,18 +41,18 @@ def _single(qid: str, task_type: str = "Single") -> GeneratedQuestion: def _pair(pid: str, task_type: str = "AR") -> list[GeneratedQuestion]: """构造一个合法孪生对(original + mirror),共享 pair_id / unit_id / flip_axis。""" - base = dict( - video_id="v", - task_type=task_type, - question="?", - options=("A", "B", "C", "D"), - answer="A", - source_nodes=(), - difficulty="hard", - pair_id=pid, - unit_id=pid, - flip_axis="before_after", - ) + base = { + "video_id": "v", + "task_type": task_type, + "question": "?", + "options": ("A", "B", "C", "D"), + "answer": "A", + "source_nodes": (), + "difficulty": "hard", + "pair_id": pid, + "unit_id": pid, + "flip_axis": "before_after", + } return [ GeneratedQuestion(question_id=f"{pid}_o", question_role="pair_original", **base), GeneratedQuestion(question_id=f"{pid}_m", question_role="pair_mirror", **base),