43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
"""ClassificationAgent：零散档案启发式与分类 none 语义（纯函数/无 LLM）。"""
|
|||
|
|
|
|||
|
|
import pytest
|
|||
|
|
|
|||
|
|
from app.agents.memoir.classification_agent import (
|
|||
|
|
ClassificationAgent,
|
|||
|
|
_looks_like_fragment_only,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@pytest.mark.parametrize(
|
|||
|
|
"text,expected_fragment",
|
|||
|
|
[
|
|||
|
|
("", True),
|
|||
|
|
(" ", True),
|
|||
|
|
("我1999年出生", True),
|
|||
|
|
("1999年出生。", True),
|
|||
|
|
("1999年出生！", True),
|
|||
|
|
("我是云南人", True),
|
|||
|
|
("我是北京籍。", True),
|
|||
|
|
("小学二年级那次下雨爷爷背我过河，鞋全湿了。", False),
|
|||
|
|
("我出生在农村，家里养过一头黄牛。", False),
|
|||
|
|
("我是北京人，后来去上海读了大学。", False),
|
|||
|
|
],
|
|||
|
|
)
|
|||
|
|
def test_looks_like_fragment_only(text: str, expected_fragment: bool) -> None:
|
|||
|
|
assert _looks_like_fragment_only(text) is expected_fragment
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_classify_skips_story_for_birth_year_without_llm() -> None:
|
|||
|
|
agent = ClassificationAgent()
|
|||
|
|
assert agent.classify("1999年出生", fallback_stage="childhood", llm=None) is None
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_classify_fallback_when_no_llm_and_narrative_snippet() -> None:
|
|||
|
|
agent = ClassificationAgent()
|
|||
|
|
out = agent.classify(
|
|||
|
|
"小学二年级的时候我在操场上摔了一跤，膝盖流了很多血，是老师背我去医务室的。",
|
|||
|
|
fallback_stage="childhood",
|
|||
|
|
llm=None,
|
|||
|
|
)
|
|||
|
|
assert out == "education"
|