mirror of https://github.com/Yomguithereal/fog.git
28 lines
710 B
Python
28 lines
710 B
Python
|
# =============================================================================
|
||
|
|
# Fog MinHash LSH Unit Tests
|
||
|
|
# =============================================================================
|
||
|
|
from pytest import approx
|
||
|
|
from fog.lsh import LSBMinHash
|
||
|
|
|
||
|
|
TESTS = [
|
||
|
|
('abc', '', 0),
|
||
|
|
('', 'abc', 0),
|
||
|
|
('', '', 1),
|
||
|
|
('abc', 'abc', 1),
|
||
|
|
('abc', 'xyz', 0),
|
||
|
|
('night', 'nacht', 3 / 7),
|
||
|
|
('context', 'contact', 4 / 7),
|
||
|
|
('ht', 'nacht', 2 / 5)
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class TestLSBMinHash(object):
|
||
|
|
def test_basics(self):
|
||
|
|
m = LSBMinHash(precision=16, seed=123)
|
||
|
|
|
||
|
|
for A, B, j in TESTS:
|
||
|
|
sA = m.hash(A)
|
||
|
|
sB = m.hash(B)
|
||
|
|
|
||
|
|
assert m.similarity(sA, sB) == approx(j, abs=1e-1)
|