gr-analytics 0.3.0__tar.gz → 0.3.1__tar.gz

{gr_analytics-0.3.0 → gr_analytics-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gr_analytics
-Version: 0.3.0
+Version: 0.3.1
 Summary: Scoring and salary calculation for GridRival fantasy F1
 Author: nce8
 License-Expression: MIT

{gr_analytics-0.3.0 → gr_analytics-0.3.1}/gr_analytics/__init__.py

@@ -722,4 +722,18 @@ def optimal_lineup(
             best_total = total
             best_full = full
 
+    # When balance == 0 the star bonus is zero so the loop picks
+    # arbitrarily.  Override to star the highest-points eligible driver.
+    if balance == 0.0:
+        best_full["star"] = 0
+        eligible = best_full[best_full["type"] == "driver"]
+        if star_salary_cap is not None:
+            eligible = eligible[eligible["starting_salary"] <= star_salary_cap]
+        if not eligible.empty:
+            best_full.loc[eligible["points_earned"].idxmax(), "star"] = 1
+
+    # Double points_earned for the starred driver so the returned
+    # DataFrame reflects actual fantasy points with star doubling.
+    best_full.loc[best_full["star"] == 1, "points_earned"] *= 2
+
     return best_full.drop(columns=["_locked"])

{gr_analytics-0.3.0 → gr_analytics-0.3.1}/gr_analytics/data/driver_data.csv

@@ -130,4 +130,37 @@ team,AMR,AMR,,3,12.8,,69
 team,WIL,WIL,,3,12.7,,80
 team,HAS,HAS,,3,11.2,,114
 team,AUD,AUD,,3,10.8,,92
-team,CAD,CAD,,3,8.4,,80
+team,CAD,CAD,,3,8.4,,80
+driver,ANT,K. Antonelli,MER,4,28.8,4,176
+driver,RUS,G. Russell,MER,4,28.7,3,165
+driver,NOR,L. Norris,MCL,4,26.7,6,136
+driver,LEC,C. Leclerc,FER,4,25.9,6,161
+driver,PIA,O. Piastri,MCL,4,25.7,8,114
+driver,VER,M. Verstappen,RBR,4,25.3,5,144
+driver,HAM,L. Hamilton,FER,4,23.3,7,157
+driver,GAS,P. Gasly,ALP,4,19.7,11,130
+driver,SAI,C. Sainz Jr,WIL,4,16.4,12,122
+driver,LAW,L. Lawson,RBS,4,16,13,126
+driver,HAD,I. Hadjar,RBR,4,15.7,13,100
+driver,ALO,F. Alonso,AMR,4,15.1,12,85
+driver,OCO,E. Ocon,HAS,4,13.2,15,127
+driver,BEA,O. Bearman,HAS,4,12.9,15,140
+driver,BOR,G. Bortoleto,AUD,4,12.7,15,105
+driver,ALB,A. Albon,WIL,4,11.9,15,87
+driver,HUL,N. Hulkenberg,AUD,4,11.3,16,91
+driver,COL,F. Colapinto,ALP,4,11,16,135
+driver,LIN,A. Lindblad,RBS,4,10.5,16,125
+driver,STR,L. Stroll,AMR,4,10.4,15,74
+driver,PER,S. Perez,CAD,4,8.1,18,98
+driver,BOT,V. Bottas,CAD,4,6,19,85
+team,MER,MER,,4,29.4,,174
+team,MCL,MCL,,4,24.7,,120
+team,FER,FER,,4,24.6,,158
+team,RBR,RBR,,4,20,,111
+team,ALP,ALP,,4,18,,118
+team,RBS,RBS,,4,15.3,,113
+team,WIL,WIL,,4,15,,90
+team,HAS,HAS,,4,12.6,,113
+team,AMR,AMR,,4,11.3,,74
+team,AUD,AUD,,4,11,,92
+team,CAD,CAD,,4,7.3,,80

{gr_analytics-0.3.0 → gr_analytics-0.3.1}/gr_analytics.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gr_analytics
-Version: 0.3.0
+Version: 0.3.1
 Summary: Scoring and salary calculation for GridRival fantasy F1
 Author: nce8
 License-Expression: MIT

{gr_analytics-0.3.0 → gr_analytics-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "gr_analytics"
-version = "0.3.0"
+version = "0.3.1"
 description = "Scoring and salary calculation for GridRival fantasy F1"
 readme = "README.md"
 requires-python = ">=3.9"

{gr_analytics-0.3.0 → gr_analytics-0.3.1}/tests/test_scoring.py

@@ -647,17 +647,25 @@ class TestOptimizeForBalance:
         assert bal_abbrs == sal_abbrs
 
     def test_balance_star_assigned_when_balance_positive(self, balance_pool):
-        """When optimize_for is a positive float, a star driver should be assigned."""
+        """When optimize_for is a positive float, a star driver should be assigned
+        and their points_earned should be doubled."""
         result = optimal_lineup(balance_pool, optimize_for=0.5)
         assert (result["star"] == 1).sum() == 1
-
-    def test_balance_0_star_has_no_effect(self, balance_pool):
-        """When optimize_for=0.0, a star may be set but has no impact on lineup choice."""
-        result_sal = optimal_lineup(balance_pool, optimize_for="salary_change")
-        result_bal = optimal_lineup(balance_pool, optimize_for=0.0)
-        sal_abbrs = sorted(result_sal["driver_abbr"].dropna().tolist())
-        bal_abbrs = sorted(result_bal["driver_abbr"].dropna().tolist())
-        assert bal_abbrs == sal_abbrs
+        star_row = result[result["star"] == 1].iloc[0]
+        orig_pts = balance_pool.loc[
+            balance_pool["driver_abbr"] == star_row["driver_abbr"], "points_earned"
+        ].iloc[0]
+        assert star_row["points_earned"] == orig_pts * 2
+
+    def test_balance_0_star_assigned_to_highest_pts(self, balance_pool):
+        """When balance=0, the highest-points eligible driver is still starred
+        and their points_earned is doubled."""
+        result = optimal_lineup(balance_pool, optimize_for=0.0)
+        assert (result["star"] == 1).sum() == 1
+        star_row = result[result["star"] == 1].iloc[0]
+        # D1 has the highest points (200) in the pool
+        assert star_row["driver_abbr"] == "D1"
+        assert star_row["points_earned"] == 400.0
 
     def test_balance_out_of_range_raises(self, balance_pool):
         """optimize_for outside [0, 1] should raise ValueError."""