Adds calculate_game_importance() that boosts Priority for high-stakes regular-season matchups based on season progress (sharp ramp after game 55), playoff bubble proximity (wildcard rank ~17-19 = max relevance), and divisional/conference rivalry (1.4x/1.2x multipliers). Max bonus 150 pts applied to both LIVE and PRE games; playoff and FINAL games are unaffected. Extends standings schema with division, conference, games_played, and wildcard_sequence fields fetched from the NHL API. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
99 lines
2.8 KiB
Python
99 lines
2.8 KiB
Python
import json
|
|
import sqlite3
|
|
|
|
import pytest
|
|
|
|
|
|
def make_game(
|
|
game_state="LIVE",
|
|
home_name="Maple Leafs",
|
|
away_name="Bruins",
|
|
home_score=2,
|
|
away_score=1,
|
|
period=3,
|
|
seconds_remaining=300,
|
|
in_intermission=False,
|
|
start_time_utc="2024-04-10T23:00:00Z",
|
|
home_record="40-25-10",
|
|
away_record="38-27-09",
|
|
game_type=2,
|
|
situation=None,
|
|
):
|
|
clock = {
|
|
"timeRemaining": f"{seconds_remaining // 60:02d}:{seconds_remaining % 60:02d}",
|
|
"secondsRemaining": seconds_remaining,
|
|
"running": game_state == "LIVE",
|
|
"inIntermission": in_intermission,
|
|
}
|
|
return {
|
|
"gameState": game_state,
|
|
"startTimeUTC": start_time_utc,
|
|
"periodDescriptor": {"number": period},
|
|
"clock": clock,
|
|
"homeTeam": {
|
|
"name": {"default": home_name},
|
|
"score": home_score,
|
|
"sog": 15,
|
|
"logo": "https://example.com/home.png",
|
|
"record": home_record,
|
|
},
|
|
"awayTeam": {
|
|
"name": {"default": away_name},
|
|
"score": away_score,
|
|
"sog": 12,
|
|
"logo": "https://example.com/away.png",
|
|
"record": away_record,
|
|
},
|
|
"gameOutcome": {"lastPeriodType": "REG"},
|
|
"gameType": game_type,
|
|
**({"situation": situation} if situation is not None else {}),
|
|
}
|
|
|
|
|
|
LIVE_GAME = make_game()
|
|
PRE_GAME = make_game(
|
|
game_state="FUT", home_score=0, away_score=0, period=0, seconds_remaining=1200
|
|
)
|
|
FINAL_GAME = make_game(game_state="OFF", period=3, seconds_remaining=0)
|
|
|
|
SAMPLE_SCOREBOARD = {"games": [LIVE_GAME, PRE_GAME, FINAL_GAME]}
|
|
|
|
|
|
@pytest.fixture()
|
|
def sample_scoreboard():
|
|
return SAMPLE_SCOREBOARD
|
|
|
|
|
|
@pytest.fixture()
|
|
def flask_client(tmp_path, monkeypatch):
|
|
data_dir = tmp_path / "data"
|
|
data_dir.mkdir()
|
|
|
|
# Write sample scoreboard JSON
|
|
scoreboard_file = data_dir / "scoreboard_data.json"
|
|
scoreboard_file.write_text(json.dumps(SAMPLE_SCOREBOARD))
|
|
|
|
# Create minimal SQLite DB so get_team_standings doesn't crash
|
|
db_path = data_dir / "nhl_standings.db"
|
|
conn = sqlite3.connect(str(db_path))
|
|
conn.execute(
|
|
"CREATE TABLE standings "
|
|
"(team_common_name TEXT, league_sequence INTEGER, league_l10_sequence INTEGER, "
|
|
"division_abbrev TEXT, conference_abbrev TEXT, games_played INTEGER, wildcard_sequence INTEGER)"
|
|
)
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
# Patch module-level path constants so no reloads are needed
|
|
import app.routes as routes
|
|
import app.games as games
|
|
|
|
monkeypatch.setattr(routes, "SCOREBOARD_DATA_FILE", str(scoreboard_file))
|
|
monkeypatch.setattr(games, "DB_PATH", str(db_path))
|
|
|
|
from app import app as flask_app
|
|
|
|
flask_app.config["TESTING"] = True
|
|
with flask_app.test_client() as client:
|
|
yield client
|