80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
import pytest
|
|
from sqlmodel import Session
|
|
|
|
from command_logic.logic_gameplay import advance_runners, get_obc, get_re24, get_wpa, complete_play
|
|
from tests.factory import session_fixture, Game
|
|
|
|
|
|
def test_advance_runners(session: Session):
|
|
this_game = session.get(Game, 3)
|
|
play_1 = this_game.initialize_play(session)
|
|
|
|
assert play_1.starting_outs == 0
|
|
assert play_1.on_first_id is None
|
|
assert play_1.on_second_id is None
|
|
assert play_1.on_third_id is None
|
|
|
|
# TODO: Test advance runners once "advance play" function is ready
|
|
|
|
def test_get_obc():
|
|
assert get_obc() == 0
|
|
assert get_obc(on_first=True) == 1
|
|
assert get_obc(on_second=True) == 2
|
|
assert get_obc(on_third=True) == 3
|
|
assert get_obc(on_first=True, on_second=True) == 4
|
|
assert get_obc(on_first=True, on_third=True) == 5
|
|
assert get_obc(on_second=True, on_third=True) == 6
|
|
assert get_obc(on_first=True, on_second=True, on_third=True) == 7
|
|
|
|
|
|
def test_get_re24(session: Session):
|
|
game_1 = session.get(Game, 1)
|
|
this_play = game_1.current_play_or_none(session)
|
|
|
|
assert this_play is not None
|
|
assert get_re24(this_play, runs_scored=0, new_obc=0, new_starting_outs=2) == -.154
|
|
assert get_re24(this_play, runs_scored=1, new_obc=0, new_starting_outs=1) == 1
|
|
assert get_re24(this_play, runs_scored=0, new_obc=2, new_starting_outs=1) == 0.365
|
|
assert get_re24(this_play, runs_scored=0, new_obc=6, new_starting_outs=2) == 0.217
|
|
|
|
|
|
def test_get_wpa(session: Session):
|
|
game_1 = session.get(Game, 1)
|
|
next_play = game_1.current_play_or_none(session) # Starting wpa: 0.564
|
|
this_play = game_1.plays[0] # Starting wpa: 0.500
|
|
|
|
assert this_play != next_play
|
|
assert get_wpa(this_play, next_play) == -0.064
|
|
|
|
next_play.starting_outs = 2
|
|
next_play.away_score = 3 # Starting wpa: 0.347
|
|
|
|
assert get_wpa(this_play, next_play) == 0.153
|
|
|
|
|
|
def test_complete_play(session: Session):
|
|
game_1 = session.get(Game, 1)
|
|
this_play = game_1.current_play_or_none(session)
|
|
|
|
assert this_play.inning_half == 'top'
|
|
assert this_play.inning_num == 1
|
|
assert this_play.starting_outs == 1
|
|
|
|
this_play.pa, this_play.ab, this_play.so, this_play.outs = 1, 1, 1, 1
|
|
next_play = complete_play(session, this_play)
|
|
|
|
assert next_play.inning_half == 'top'
|
|
assert this_play.inning_num == 1
|
|
assert next_play.starting_outs == 2
|
|
assert next_play.is_tied
|
|
assert next_play.managerai == this_play.managerai
|
|
assert this_play.re24 == -0.154
|
|
|
|
next_play.pa, next_play.ab, next_play.double, next_play.batter_final = 1, 1, 1, 2
|
|
third_play = complete_play(session, next_play)
|
|
|
|
assert third_play.on_base_code == 2
|
|
assert next_play.re24 == 0.182
|
|
|
|
|