81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
"""数据加载器测试"""
|
|
|
|
import sys
|
|
import os
|
|
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", ".."))
|
|
|
|
from data.crowd_db.loader import CrowdDBLoader, CrowdRecommendation
|
|
|
|
|
|
def test_load_hunan_data():
|
|
"""测试加载湖南数据"""
|
|
loader = CrowdDBLoader()
|
|
data = loader.load_province("湖南")
|
|
assert data is not None
|
|
assert data["province"] == "湖南"
|
|
assert len(data["score_ranges"]) > 0
|
|
|
|
|
|
def test_find_recommendations_in_range():
|
|
"""测试查询分数段内的推荐"""
|
|
loader = CrowdDBLoader()
|
|
recs = loader.find_recommendations("湖南", score=575)
|
|
assert isinstance(recs, list)
|
|
# 575分应该在 560-580 范围内
|
|
if recs:
|
|
assert all(r["frequency"] > 0 for r in recs)
|
|
|
|
|
|
def test_find_recommendations_by_school():
|
|
"""测试按院校名查询推荐"""
|
|
loader = CrowdDBLoader()
|
|
rec = loader.find_recommendation_by_school("湖南", "长沙理工大学")
|
|
assert rec is not None
|
|
assert rec["name"] == "长沙理工大学"
|
|
|
|
|
|
def test_load_nonexistent_province():
|
|
"""测试加载不存在的省份"""
|
|
loader = CrowdDBLoader()
|
|
data = loader.load_province("不存在的省")
|
|
assert data is None
|
|
|
|
|
|
def test_crowd_recommendation_dataclass():
|
|
"""测试数据类"""
|
|
rec = CrowdRecommendation(
|
|
name="测试大学",
|
|
major="测试专业",
|
|
frequency=4,
|
|
platforms=["千问", "元宝", "百度", "豆包"],
|
|
predicted_increase=15,
|
|
alternatives=[],
|
|
)
|
|
assert rec.frequency == 4
|
|
assert rec.risk_level == "high" # frequency=4 应该是高风险
|
|
|
|
|
|
def test_crowd_recommendation_risk_level_full_range():
|
|
"""验证 CrowdRecommendation.risk_level 在 frequency 0-5 全段与 detector 一致
|
|
|
|
频率边界（与 crowd_detector._risk_level_from_frequency 保持一致）：
|
|
- 0 → 'none' (不构成风险)
|
|
- 1 → 'low'
|
|
- 2, 3 → 'medium'
|
|
- 4, 5 → 'high'
|
|
"""
|
|
expected = {0: "none", 1: "low", 2: "medium", 3: "medium", 4: "high", 5: "high"}
|
|
for freq, want in expected.items():
|
|
rec = CrowdRecommendation(
|
|
name="x",
|
|
major="y",
|
|
frequency=freq,
|
|
platforms=[],
|
|
predicted_increase=0,
|
|
alternatives=[],
|
|
)
|
|
assert rec.risk_level == want, (
|
|
f"frequency={freq} → expected '{want}', got '{rec.risk_level}'"
|
|
)
|