mirror of
https://github.com/PR0M3TH3AN/unredact.git
synced 2026-10-27 08:40:29 +00:00
Alex-GilbertandClaude Opus 4.6
b357713e5e
Add solve_dictionary() for linear-scan matching of dictionary entries against a target pixel width, and DictionaryStore for managing named word/name lists. Reuses SolveResult from the solver module. Co-Authored-By: Claude Opus 4.6 <[email protected]>
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
# tests/test_dictionary.py
|
|
import pytest
|
|
from PIL import ImageFont
|
|
|
|
from unredact.pipeline.dictionary import solve_dictionary, DictionaryStore
|
|
|
|
|
|
def _get_test_font() -> ImageFont.FreeTypeFont:
|
|
import subprocess
|
|
result = subprocess.run(
|
|
["fc-match", "--format=%{file}", "Liberation Serif"],
|
|
capture_output=True, text=True,
|
|
)
|
|
path = result.stdout.strip()
|
|
return ImageFont.truetype(path, 40)
|
|
|
|
|
|
class TestSolveDictionary:
|
|
def test_finds_matching_word(self):
|
|
font = _get_test_font()
|
|
entries = ["Smith", "Jones", "Brown"]
|
|
target = font.getlength("Smith")
|
|
results = solve_dictionary(font, entries, target, tolerance=0.5)
|
|
texts = [r.text for r in results]
|
|
assert "Smith" in texts
|
|
|
|
def test_respects_tolerance(self):
|
|
font = _get_test_font()
|
|
entries = ["Smith", "Jones", "Brown"]
|
|
target = font.getlength("Smith")
|
|
results = solve_dictionary(font, entries, target, tolerance=0.0)
|
|
assert all(r.error == 0.0 for r in results)
|
|
|
|
def test_with_context(self):
|
|
font = _get_test_font()
|
|
entries = ["Smith"]
|
|
target = font.getlength(" Smith ") - font.getlength(" ") - font.getlength(" ")
|
|
results = solve_dictionary(
|
|
font, entries, target, tolerance=1.0,
|
|
left_context=" ", right_context=" ",
|
|
)
|
|
assert len(results) >= 0 # may or may not match due to kerning
|
|
|
|
|
|
class TestDictionaryStore:
|
|
def test_add_and_list(self):
|
|
store = DictionaryStore()
|
|
store.add("names", ["Alice", "Bob", "Charlie"])
|
|
assert "names" in store.list()
|
|
|
|
def test_get_entries(self):
|
|
store = DictionaryStore()
|
|
store.add("names", ["Alice", "Bob"])
|
|
assert store.get_entries("names") == ["Alice", "Bob"]
|
|
|
|
def test_remove(self):
|
|
store = DictionaryStore()
|
|
store.add("names", ["Alice"])
|
|
store.remove("names")
|
|
assert "names" not in store.list()
|
|
|
|
def test_all_entries(self):
|
|
store = DictionaryStore()
|
|
store.add("list1", ["Alice", "Bob"])
|
|
store.add("list2", ["Charlie"])
|
|
all_entries = store.all_entries()
|
|
assert set(all_entries) == {"Alice", "Bob", "Charlie"}
|