Use hypothesis.assume to filter out uninteresting test data

FAForever · Jul 13, 2021 · 679dc35 · 679dc35
1 parent 88de96e
commit 679dc35
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tests/unit_tests/test_matchmaker_algorithm_bucket_teams.py b/tests/unit_tests/test_matchmaker_algorithm_bucket_teams.py
@@ -2,7 +2,7 @@
 import random
 
 import pytest
-from hypothesis import given
+from hypothesis import assume, given
 from hypothesis import strategies as st
 
 from server import config
@@ -45,6 +45,8 @@ def make(
 def test_make_teams_single_correct_size(searches, size, make_teams_func):
     matched, _ = make_teams_func(searches, size)
 
+    assume(matched != [])
+
     for search in matched:
         assert len(search.players) == size
 

diff --git a/tests/unit_tests/test_matchmaker_algorithm_stable_marriage.py b/tests/unit_tests/test_matchmaker_algorithm_stable_marriage.py
@@ -1,7 +1,7 @@
 import logging
 
 import pytest
-from hypothesis import given, settings
+from hypothesis import assume, given, settings
 
 from server import config
 from server.matchmaker import Search
@@ -168,6 +168,8 @@ def test_stable_marriage_produces_symmetric_matchings(
 
         matches = stable_marriage.StableMarriage().find(ranks)
 
+        assume(matches != [])
+
         for search in matches:
             opponent = matches[search]
             assert matches[opponent] == search