Spaces:
Sleeping
Sleeping
File size: 5,151 Bytes
0dfe795 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 | from __future__ import annotations
import os
import sys
import unittest
import pandas as pd
sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
from data.mlb_starters import (
build_oddsapi_starter_fallback_map,
merge_probable_starters_with_odds_fallback,
)
class TestMlbStarters(unittest.TestCase):
def test_oddsapi_fallback_can_fill_tba_matchup_from_pitcher_team_inference(self) -> None:
props_feed = pd.DataFrame(
[
{
"event_id": "event-1",
"away_team": "New York Yankees",
"home_team": "San Francisco Giants",
"market_family": "k",
"selection_scope": "pitcher",
"player_name_raw": "Max Fried",
},
{
"event_id": "event-1",
"away_team": "New York Yankees",
"home_team": "San Francisco Giants",
"market_family": "k",
"selection_scope": "pitcher",
"player_name_raw": "Logan Webb",
},
]
)
pitcher_statcast_df = pd.DataFrame(
[
{
"player_name": "Max Fried",
"inning_topbot": "Bottom",
"away_team": "New York Yankees",
"home_team": "San Francisco Giants",
},
{
"player_name": "Logan Webb",
"inning_topbot": "Top",
"away_team": "New York Yankees",
"home_team": "San Francisco Giants",
},
]
)
primary = {
("new york yankees", "san francisco giants"): {
"away_team_raw": "New York Yankees",
"home_team_raw": "San Francisco Giants",
"away_pitcher": None,
"home_pitcher": None,
"away_pitcher_source": "unresolved",
"home_pitcher_source": "unresolved",
"starter_cache_source": "statsapi_probable_pitcher",
"fallback_used": False,
}
}
fallback = build_oddsapi_starter_fallback_map(
props_feed=props_feed,
primary_starters=primary,
pitcher_statcast_df=pitcher_statcast_df,
)
merged = merge_probable_starters_with_odds_fallback(primary, fallback)
row = merged[("new york yankees", "san francisco giants")]
self.assertEqual(row["away_pitcher"], "Max Fried")
self.assertEqual(row["home_pitcher"], "Logan Webb")
self.assertEqual(row["away_pitcher_source"], "oddsapi_pitcher_strikeouts_two_candidate_match")
self.assertEqual(row["home_pitcher_source"], "oddsapi_pitcher_strikeouts_two_candidate_match")
self.assertEqual(row["starter_cache_source"], "statsapi_plus_oddsapi_fallback")
self.assertTrue(bool(row["fallback_used"]))
def test_oddsapi_fallback_does_not_override_existing_mlb_side(self) -> None:
props_feed = pd.DataFrame(
[
{
"event_id": "event-2",
"away_team": "Cleveland Guardians",
"home_team": "Detroit Tigers",
"market_family": "k",
"selection_scope": "pitcher",
"player_name_raw": "Tanner Bibee",
},
{
"event_id": "event-2",
"away_team": "Cleveland Guardians",
"home_team": "Detroit Tigers",
"market_family": "k",
"selection_scope": "pitcher",
"player_name_raw": "Tarik Skubal",
},
]
)
primary = {
("cleveland guardians", "detroit tigers"): {
"away_team_raw": "Cleveland Guardians",
"home_team_raw": "Detroit Tigers",
"away_pitcher": "Tanner Bibee",
"home_pitcher": None,
"away_pitcher_source": "statsapi_probable_pitcher",
"home_pitcher_source": "unresolved",
"starter_cache_source": "statsapi_probable_pitcher",
"fallback_used": False,
}
}
fallback = build_oddsapi_starter_fallback_map(
props_feed=props_feed,
primary_starters=primary,
pitcher_statcast_df=None,
)
merged = merge_probable_starters_with_odds_fallback(primary, fallback)
row = merged[("cleveland guardians", "detroit tigers")]
self.assertEqual(row["away_pitcher"], "Tanner Bibee")
self.assertEqual(row["away_pitcher_source"], "statsapi_probable_pitcher")
self.assertEqual(row["home_pitcher"], "Tarik Skubal")
self.assertEqual(row["home_pitcher_source"], "oddsapi_pitcher_strikeouts_single_candidate_match")
self.assertEqual(row["starter_cache_source"], "statsapi_plus_oddsapi_fallback")
if __name__ == "__main__":
unittest.main()
|