| """ |
| Tests for the Object Mapper — synonym normalization and fuzzy matching. |
| """ |
|
|
| import pytest |
|
|
| from src.utils.object_mapper import ( |
| normalize_label, |
| find_best_match, |
| SYNONYM_MAP, |
| ) |
|
|
|
|
| class TestNormalizeLabel: |
| """Tests for normalize_label().""" |
|
|
| def test_known_synonym(self): |
| assert normalize_label("mug") == "cup" |
|
|
| def test_known_synonym_case_insensitive(self): |
| assert normalize_label("MUG") == "cup" |
|
|
| def test_unknown_label_passthrough(self): |
| assert normalize_label("bottle") == "bottle" |
|
|
| def test_whitespace_stripped(self): |
| assert normalize_label(" mug ") == "cup" |
|
|
| def test_container_to_bottle(self): |
| assert normalize_label("container") == "bottle" |
|
|
| def test_smartphone_to_cell_phone(self): |
| assert normalize_label("smartphone") == "cell phone" |
|
|
|
|
| class TestFindBestMatch: |
| """Tests for find_best_match().""" |
|
|
| @pytest.fixture |
| def detected_labels(self) -> list[str]: |
| return ["bottle", "cup", "box", "person", "laptop"] |
|
|
| def test_exact_match(self, detected_labels): |
| result = find_best_match("bottle", detected_labels) |
| assert result == "bottle" |
|
|
| def test_synonym_match(self, detected_labels): |
| """'mug' should resolve to 'cup' via synonym normalization.""" |
| result = find_best_match("mug", detected_labels) |
| assert result == "cup" |
|
|
| def test_container_matches_bottle(self, detected_labels): |
| result = find_best_match("container", detected_labels) |
| assert result == "bottle" |
|
|
| def test_substring_match(self, detected_labels): |
| result = find_best_match("lap", detected_labels) |
| assert result == "laptop" |
|
|
| def test_no_match_returns_none(self, detected_labels): |
| result = find_best_match("xyznonexistent", detected_labels) |
| assert result is None |
|
|
| def test_empty_query(self, detected_labels): |
| result = find_best_match("", detected_labels) |
| |
| assert result is None or isinstance(result, str) |
|
|
| def test_empty_detected_labels(self): |
| result = find_best_match("bottle", []) |
| assert result is None |
|
|
