Jesse_Chen
48 lines
2.5 KiB
Python
48 lines
2.5 KiB
Python
from datetime import date
|
|
import unittest
|
|
|
|
from scripts.rectification.contracts import normalize_rectification_request
|
|
from scripts.rectification.scoring_service import build_event_contribution_matrix, calculation_spec, sha256
|
|
|
|
EVENT_ID = "00000000-0000-4000-8000-000000000002"
|
|
|
|
def request():
|
|
return {
|
|
"birth_date": "1990-01-02", "start_time": "05:00", "end_time": "06:00",
|
|
"lat": 25.033, "lon": 121.5654, "tz": 8,
|
|
"events": [{"id": EVENT_ID, "domain": "career", "event_kind": "career_change", "date_start": "2020-01-01", "date_end": "2020-12-31", "precision": "year", "summary": "工作变动"}],
|
|
}
|
|
|
|
class RectificationProvenanceTest(unittest.TestCase):
|
|
def test_missing_and_explicit_null_remain_distinct(self):
|
|
legacy = normalize_rectification_request(request(), today=date(2026, 7, 30))
|
|
self.assertNotIn("birth_time_source", legacy)
|
|
enriched_input = request()
|
|
enriched_input["birth_time_source"] = None
|
|
enriched_input["events"][0]["date_source"] = None
|
|
enriched = normalize_rectification_request(enriched_input, today=date(2026, 7, 30))
|
|
self.assertIn("birth_time_source", enriched)
|
|
self.assertIsNone(enriched["birth_time_source"])
|
|
self.assertIn("date_source", enriched["events"][0])
|
|
|
|
def test_enriched_spec_hash_matches_typescript_fixture(self):
|
|
value = request()
|
|
value.update({"birth_time_source": "family_exact", "timezone_id": "Asia/Taipei", "timezone_source": "iana_historical", "local_time_status": "resolved"})
|
|
normalized = normalize_rectification_request(value, today=date(2026, 7, 30))
|
|
self.assertEqual(sha256(calculation_spec(normalized)), "fa4afe79228bedebd809c7b3c9d9d32a428f7066e3e1fd1813e44b317bd38e66")
|
|
|
|
def test_event_provenance_is_not_forwarded_to_scoring_weights(self):
|
|
value = request()
|
|
value["events"][0].update({"date_source": "user_reported", "date_reliability": "medium"})
|
|
normalized = normalize_rectification_request(value, today=date(2026, 7, 30))
|
|
seen = []
|
|
def provider(payload):
|
|
seen.append(payload)
|
|
return [{"time": "05:00", "score": 1, "evidence": [{"event_id": EVENT_ID, "domain": "career", "candidate_time": "05:00", "rule_ids": ["D10:test"], "points": 1}], "missing_layers": []}]
|
|
build_event_contribution_matrix(normalized, provider)
|
|
self.assertTrue(seen)
|
|
self.assertTrue(all("date_source" not in payload["events"][0] for payload in seen))
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|