Better test

BenBrostoff · Jan 25, 2024 · 8a083d9 · 8a083d9
1 parent eec49ac
commit 8a083d9
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/draftfast/test/test_nhl.py b/draftfast/test/test_nhl.py
@@ -37,12 +37,16 @@ def test_nhl_dk_unsolvable_three_team_non_goalie_restriction():
     unique_teams = set()
     new_player_pool = []
     for p in player_pool:
-        if p.pos != 'G':
-            unique_teams.add(p.team)
         if len(unique_teams) > 2 and p.team not in unique_teams:
             continue
 
-        new_player_pool.append(p)
+        if p.pos != 'G':
+            unique_teams.add(p.team)
+
+        if len(unique_teams) < 3:
+            new_player_pool.append(p)
+
+    assertions.assertEqual(len(unique_teams) > 2, True)
 
     roster = run(
         rule_set=rules.DK_NHL_RULE_SET,