Fix min teams for all DK

BenBrostoff · Dec 26, 2023 · 9912072 · 9912072
1 parent ba2ebc6
commit 9912072
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/draftfast/rules.py b/draftfast/rules.py
@@ -50,6 +50,7 @@ def __init__(
     salary_max=SALARY_CAP_BY_SITE_BY_LEAGUE[DRAFT_KINGS]["NBA"],
     position_limits=POSITIONS_BY_SITE_BY_LEAGUE[DRAFT_KINGS]["NBA"],
     general_position_limits=NBA_GENERAL_POSITIONS,
+    min_teams=3,
 )
 
 DK_NBA_SHOWDOWN_RULE_SET = RuleSet(
@@ -78,6 +79,7 @@ def __init__(
     salary_max=SALARY_CAP_BY_SITE_BY_LEAGUE[DRAFT_KINGS]["WNBA"],
     position_limits=POSITIONS_BY_SITE_BY_LEAGUE[DRAFT_KINGS]["WNBA"],
     general_position_limits=NBA_GENERAL_POSITIONS,
+    min_teams=3,
 )
 
 FD_WNBA_RULE_SET = RuleSet(
@@ -98,6 +100,7 @@ def __init__(
     offensive_positions=["QB", "RB", "WR", "TE"],
     defensive_positions=["DST"],
     general_position_limits=[],
+    min_teams=3,
 )
 
 FD_NFL_RULE_SET = RuleSet(
@@ -253,6 +256,7 @@ def __init__(
     position_limits=POSITIONS_BY_SITE_BY_LEAGUE[DRAFT_KINGS]["NHL"],
     offensive_positions=["C", "W"],
     defensive_positions=["G", "D"],
+    min_teams=3,
     general_position_limits=[],
 )
 

diff --git a/draftfast/test/test_optimize.py b/draftfast/test/test_optimize.py
@@ -739,15 +739,16 @@ def test_no_opposing_def_dk_nfl_mock():
     assertions.assertEqual(roster, None)
 
     # relax min teams
+    rules.DK_NFL_RULE_SET.min_teams = 1
     roster = run(
         rule_set=rules.DK_NFL_RULE_SET,
         player_pool=mock_pool,
         optimizer_settings=OptimizerSettings(
-            min_teams=1,
             no_offense_against_defense=True,
         ),
         verbose=True,
     )
+    rules.DK_NFL_RULE_SET.min_teams = 3
 
     assertions.assertEqual(roster.projected(), 877)
     assertions.assertEqual(len(set([p.team for p in roster.players])), 1)