You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
739 B
Python
25 lines
739 B
Python
import pytest
|
|
|
|
from app.llm.heuristic import HeuristicProvider
|
|
|
|
CTX = {
|
|
"projects": [
|
|
{"id": "life-trip", "name": "여행 — 한국", "folder_id": "life"},
|
|
{"id": "life-fam", "name": "가족", "folder_id": "life"},
|
|
{"id": "onb", "name": "온보딩 리디자인", "folder_id": "work"},
|
|
]
|
|
}
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"raw,ctype",
|
|
[
|
|
("수요일 11시 자전거 수리 맡기기", "event"), # 시각 명시
|
|
("다음 주에 비행기 티켓 사기", "task"), # 행동+기한
|
|
("온보딩에 애니메이션 넣으면 어떨까", "idea"), # 막연
|
|
],
|
|
)
|
|
def test_heuristic_type(raw, ctype):
|
|
c = HeuristicProvider().classify_capture(raw, CTX)
|
|
assert c.type == ctype
|