Assert v11.1 candidate categories

tests/test_candidate_scanner_regression.py

@@ -2,6 +2,7 @@ from candidate_scanner import scan_unmasked_candidates
 from test_cases.legal_regression_cases import (
     CANDIDATE_SCANNER_TEXT,
     EXPECTED_CANDIDATES,
+    EXPECTED_CANDIDATE_TYPES,
     NEGATIVE_VALUES,
 )
 
@@ -14,6 +15,18 @@ def test_candidate_scanner_finds_expected_reference_like_values():
     assert not missing, f"Expected candidates were not found: {sorted(missing)}"
 
 
+def test_candidate_scanner_assigns_expected_reference_categories():
+    rows = scan_unmasked_candidates(CANDIDATE_SCANNER_TEXT, analyzer_results=[])
+    by_text = {row["text"]: row["entity_type"] for row in rows}
+
+    mismatches = {
+        value: {"expected": expected_type, "actual": by_text.get(value)}
+        for value, expected_type in EXPECTED_CANDIDATE_TYPES.items()
+        if by_text.get(value) != expected_type
+    }
+    assert not mismatches, f"Candidate type mismatches: {mismatches}"
+
+
 def test_candidate_scanner_does_not_suggest_known_false_positives():
     rows = scan_unmasked_candidates(CANDIDATE_SCANNER_TEXT, analyzer_results=[])
     found = {row["text"] for row in rows}
@@ -29,9 +42,11 @@ def test_candidate_scanner_preserves_context_by_returning_value_only():
     assert "reparatienummer REP-2026-4410" not in found
     assert "intern incidentnummer INC-2026-0912" not in found
     assert "kenteken XX123X" not in found
+    assert "KvK-nummer vennootschap: 76543210" not in found
     assert "REP-2026-4410" in found
     assert "INC-2026-0912" in found
     assert "XX123X" in found
+    assert "76543210" in found
 
 
 def test_candidate_scanner_respects_existing_analyzer_spans():