fix test

rapidfuzz · Nov 1, 2023 · b48575f · b48575f
1 parent 35d3b9e
commit b48575f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/tests/test_JaroWinkler.py b/tests/test_JaroWinkler.py
@@ -16,7 +16,7 @@ def test_hash_special_case(self):
         self._jaro_winkler_similarity([0, -1], [0, -2], 0.66666)
 
     def test_edge_case_lengths(self):
-        self._jaro_winkler_similarity("", "", 0)
+        self._jaro_winkler_similarity("", "", 1)
         self._jaro_winkler_similarity("0", "0", 1)
         self._jaro_winkler_similarity("00", "00", 1)
         self._jaro_winkler_similarity("0", "00", 0.85)

diff --git a/tests/test_hypothesis.py b/tests/test_hypothesis.py
@@ -10,8 +10,11 @@ def isclose(a, b, rel_tol=1e-09, abs_tol=0.0):
 
 
 def jaro_similarity(P, T):
-    P_flag = [0] * (len(P) + 1)
-    T_flag = [0] * (len(T) + 1)
+    if not P and not T:
+        return 1.0
+
+    P_flag = [False] * (len(P) + 1)
+    T_flag = [False] * (len(T) + 1)
 
     Bound = max(len(P), len(T)) // 2
     Bound = max(Bound - 1, 0)
@@ -23,8 +26,8 @@ def jaro_similarity(P, T):
 
         for j in range(lowlim, hilim + 1):
             if not P_flag[j] and P[j] == T[i]:
-                T_flag[i] = 1
-                P_flag[j] = 1
+                T_flag[i] = True
+                P_flag[j] = True
                 CommonChars += 1
                 break