50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
from app.config import AppConfig
|
|
from app.deepseek import persist_high_confidence_mappings
|
|
from app.models import MappingDecision
|
|
import yaml
|
|
|
|
|
|
def test_high_confidence_mapping_is_accepted(tmp_path):
|
|
source = AppConfig.load()
|
|
config_path = tmp_path / "mapping.yaml"
|
|
config_path.write_text(source.path.read_text(encoding="utf-8"), encoding="utf-8")
|
|
config = AppConfig.load(config_path)
|
|
|
|
accepted = persist_high_confidence_mappings(
|
|
config,
|
|
[
|
|
MappingDecision(
|
|
source_field="LINE-NO",
|
|
target_field="line_no",
|
|
confidence=0.96,
|
|
reason="字段名和样例都表示管线号",
|
|
),
|
|
MappingDecision(
|
|
source_field="UNKNOWN-X",
|
|
target_field="diameter",
|
|
confidence=0.3,
|
|
),
|
|
],
|
|
threshold=0.85,
|
|
)
|
|
|
|
assert accepted == {"LINE-NO": "line_no"}
|
|
reloaded = AppConfig.load(config_path)
|
|
assert reloaded.dynamic_field_mappings["LINE-NO"] == "line_no"
|
|
assert "UNKNOWN-X" not in reloaded.dynamic_field_mappings
|
|
|
|
|
|
def test_deepseek_key_reads_from_config(tmp_path):
|
|
source = AppConfig.load()
|
|
config_path = tmp_path / "mapping.yaml"
|
|
data = yaml.safe_load(source.path.read_text(encoding="utf-8"))
|
|
data.setdefault("deepseek", {})["api_key"] = "test-key"
|
|
config_path.write_text(yaml.safe_dump(data, allow_unicode=True, sort_keys=False), encoding="utf-8")
|
|
config = AppConfig.load(config_path)
|
|
|
|
from app.deepseek import DeepSeekMapper
|
|
|
|
mapper = DeepSeekMapper(config)
|
|
assert mapper.api_key == "test-key"
|
|
assert mapper.enabled
|