From 5ac3223a4d8bd336379fd293fb8e7fa2e2cd33a5 Mon Sep 17 00:00:00 2001
From: Max Bachmann <oss@maxbachmann.de>
Date: Sat, 1 Apr 2023 20:40:28 +0200
Subject: [PATCH] fix generator test

---
 tests/test_process.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/test_process.py b/tests/test_process.py
index 303670ae..a2acd9be 100644
--- 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():