mirror of https://github.com/Yomguithereal/fog.git
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
from Levenshtein import distance
|
|
from fog.metrics import levenshtein_distance, limited_levenshtein_distance
|
|
from experiments.utils import Timer
|
|
|
|
BASIC_TESTS = [
|
|
('book', 'back', 2),
|
|
('bbbbookkkk', 'bbbbackkkk', 2),
|
|
('hello', 'helo', 1),
|
|
('good sir', 'baal', 8),
|
|
('say', 'shiver', 5),
|
|
('feature', 'get-project-features', 13),
|
|
('example', 'samples', 3),
|
|
('sturgeon', 'urgently', 6),
|
|
('levenshtein', 'frankenstein', 6),
|
|
('distance', 'difference', 5),
|
|
('a', 'b', 1),
|
|
('ab', 'ac', 1),
|
|
('ac', 'bc', 1),
|
|
('abc', 'axc', 1),
|
|
('xabxcdxxefxgx', '1ab2cd34ef5g6', 6),
|
|
('a', '', 1),
|
|
('ab', 'a', 1),
|
|
('ab', 'b', 1),
|
|
('abc', 'ac', 1),
|
|
('xabxcdxxefxgx', 'abcdefg', 6),
|
|
('', 'a', 1),
|
|
('a', 'ab', 1),
|
|
('b', 'ab', 1),
|
|
('ac', 'abc', 1),
|
|
('abcdefg', 'xabxcdxxefxgx', 6),
|
|
('', '', 0),
|
|
('a', 'a', 0),
|
|
('abc', 'abc', 0),
|
|
('', '', 0),
|
|
('a', '', 1),
|
|
('', 'a', 1),
|
|
('abc', '', 3),
|
|
('', 'abc', 3),
|
|
('因為我是中國人所以我會說中文', '因為我是英國人所以我會說英文', 2)
|
|
]
|
|
|
|
RUNS = 100_000
|
|
SIZE_MULTIPLICATOR = 1
|
|
|
|
with Timer('python-Levenshtein'):
|
|
for _ in range(RUNS):
|
|
for A, B, _ in BASIC_TESTS:
|
|
distance(A * SIZE_MULTIPLICATOR, B * SIZE_MULTIPLICATOR)
|
|
|
|
with Timer('fog'):
|
|
for _ in range(RUNS):
|
|
for A, B, _ in BASIC_TESTS:
|
|
levenshtein_distance(A * SIZE_MULTIPLICATOR, B * SIZE_MULTIPLICATOR)
|
|
|
|
with Timer('fog-limited'):
|
|
for _ in range(RUNS):
|
|
for A, B, _ in BASIC_TESTS:
|
|
limited_levenshtein_distance(1, A * SIZE_MULTIPLICATOR, B * SIZE_MULTIPLICATOR)
|
|
|