fix generator test

rapidfuzz · Apr 1, 2023 · 5ac3223 · 5ac3223
1 parent a7b2a39
commit 5ac3223
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_process.py b/tests/test_process.py
@@ -411,8 +411,10 @@ def generate_choices():
         yield from choices
 
     search = 'aaa'
-    result = process.extract(search, generate_choices())
-    assert len(result) > 0
+    res1 = process_cpp.extract(search, generate_choices())
+    res2 = process_py.extract(search, generate_choices())
+    assert res1 == res2
+    assert len(res1) > 0
 
 
 def test_cdist_pure_python_dtype():