Add tests for search_all with best_match

aeon0 · Jun 23, 2022 · fa84dff · fa84dff
1 parent 7dded5e
commit fa84dff
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/test/template_finder_test.py b/test/template_finder_test.py
@@ -41,4 +41,15 @@ def test_search_all():
     - test passes if 3 matches result
     """
     matches = template_finder.search_all(empty, image, threshold=0.98)
-    assert len(matches) == 3
+    assert len(matches) == 3
+    """
+    Test first / best match
+    """
+    matches = template_finder.search_all(cross, image, threshold=0.6, best_match=True)
+    assert len(matches) == 1
+    matches = template_finder.search_all([cross], image, threshold=0.6, best_match=True)
+    assert len(matches) == 1
+    matches = template_finder.search_all(slash, image, threshold=0.6, best_match=True)
+    assert len(matches) == 1
+    matches = template_finder.search_all([slash], image, threshold=0.6, best_match=True)
+    assert len(matches) == 1