diff --git a/tests/unit_tests/test_matchmaker_algorithm_bucket_teams.py b/tests/unit_tests/test_matchmaker_algorithm_bucket_teams.py index ad697c8a3..8bebebc65 100644 --- 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 index 19a4fdfae..081d90f6f 100644 --- 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