def test_normal_kick_very_sunny(board):
home_team, away_team = board.teams
replay = Replay(home_team, away_team, [], [])
cmds = iter_([
SetupCompleteCommand(1, 0, TeamType.AWAY.value, 0, []),
SetupCompleteCommand(1, 0, TeamType.HOME.value, 0, []),
KickoffCommand(1, 0, TeamType.AWAY.value, 0, [8, 15]),
PreKickoffCompleteCommand(1, 0, TeamType.HOME.value, 0, [TeamType.HOME.value])
])
log_entries = iter_([
KickDirectionLogEntry(TeamType.AWAY, "1", ScatterDirection.S.value),
KickDistanceLogEntry(TeamType.AWAY, "1", 1),
KickoffEventLogEntry(KickoffEvent.CHANGING_WEATHER.value),
WeatherLogEntry(Weather.VERY_SUNNY.name),
BounceLogEntry(ScatterDirection.N.value),
BounceLogEntry(ScatterDirection.E.value)
])
events = replay._process_kickoff(cmds, log_entries, board)
event = next(events)
assert isinstance(event, TeamSetupComplete)
assert event.team == TeamType.AWAY
event = next(events)
assert isinstance(event, TeamSetupComplete)
assert event.team == TeamType.HOME
event = next(events)
assert isinstance(event, SetupComplete)
event = next(events)
assert isinstance(event, Kickoff)
assert event.target == Position(8, 15)
assert event.scatter_direction == ScatterDirection.S
assert event.scatter_distance == 1
assert board.get_ball_position() == Position(8, 14)
event = next(events)
assert isinstance(event, KickoffEventTuple)
assert event.result == KickoffEvent.CHANGING_WEATHER
event = next(events)
assert isinstance(event, WeatherTuple)
assert event.result == Weather.VERY_SUNNY
event = next(events)
assert isinstance(event, Bounce)
assert event.scatter_direction == ScatterDirection.N
assert event.start_space == Position(8, 14)
assert event.end_space == Position(8, 15)
event = next(events)
assert isinstance(event, Bounce)
assert event.scatter_direction == ScatterDirection.E
assert event.start_space == Position(8, 15)
assert event.end_space == Position(9, 15)
assert not next(cmds, None)
assert not next(log_entries, None)
def test_failed_catch_on_kick(board):
home_team, away_team = board.teams
replay = Replay(home_team, away_team, [], [])
cmds = iter_([
SetupCompleteCommand(1, 0, TeamType.AWAY.value, 0, []),
SetupCommand(1, 0, TeamType.HOME.value, 0, [TeamType.HOME.value, 0, 8, 14]),
SetupCompleteCommand(1, 0, TeamType.HOME.value, 0, []),
KickoffCommand(1, 0, TeamType.AWAY.value, 0, [8, 15]),
PreKickoffCompleteCommand(1, 0, TeamType.HOME.value, 0, [TeamType.HOME.value])
])
log_entries = iter_([
KickDirectionLogEntry(TeamType.AWAY, "1", ScatterDirection.S.value),
KickDistanceLogEntry(TeamType.AWAY, "1", 1),
KickoffEventLogEntry(KickoffEvent.CHEERING_FANS.value),
CatchEntry(TeamType.HOME, "1", "2+", "1", ActionResult.FAILURE),
BounceLogEntry(ScatterDirection.N.value)
])
events = replay._process_kickoff(cmds, log_entries, board)
event = next(events)
assert isinstance(event, TeamSetupComplete)
assert event.team == TeamType.AWAY
event = next(events)
assert isinstance(event, TeamSetupComplete)
assert event.team == TeamType.HOME
event = next(events)
assert isinstance(event, SetupComplete)
event = next(events)
assert isinstance(event, Kickoff)
assert event.target == Position(8, 15)
assert event.scatter_direction == ScatterDirection.S
assert event.scatter_distance == 1
assert board.get_ball_position() == Position(8, 14)
event = next(events)
assert isinstance(event, KickoffEventTuple)
assert event.result == KickoffEvent.CHEERING_FANS
event = next(events)
assert isinstance(event, Action)
assert event.action == ActionType.CATCH
assert event.result == ActionResult.FAILURE
event = next(events)
assert isinstance(event, Bounce)
assert event.scatter_direction == ScatterDirection.N
assert event.start_space == Position(8, 14)
assert event.end_space == Position(8, 15)
assert board.get_ball_carrier() is None
assert not next(events, None)
assert not next(cmds, None)
assert not next(log_entries, None)
def test_touchback_for_off_pitch_bounce(board):
home_team, away_team = board.teams
replay = Replay(home_team, away_team, [], [])
player = home_team.get_player(0)
cmds = iter_([
SetupCompleteCommand(1, 0, TeamType.AWAY.value, 0, []),
SetupCompleteCommand(1, 0, TeamType.HOME.value, 0, []),
KickoffCommand(1, 0, TeamType.AWAY.value, 0, [1, 1]), # Kick right into the corner,
PreKickoffCompleteCommand(1, 0, TeamType.HOME.value, 0, [TeamType.HOME.value]),
TouchbackCommand(1, 0, TeamType.HOME.value, 0, [TeamType.HOME.value, 0])
])
log_entries = iter_([
KickDirectionLogEntry(TeamType.AWAY, "1", ScatterDirection.SW.value),
KickDistanceLogEntry(TeamType.AWAY, "1", 1),
KickoffEventLogEntry(KickoffEvent.CHEERING_FANS.value),
BounceLogEntry(ScatterDirection.S.value)
])
events = replay._process_kickoff(cmds, log_entries, board)
event = next(events)
assert isinstance(event, TeamSetupComplete)
assert event.team == TeamType.AWAY
event = next(events)
assert isinstance(event, TeamSetupComplete)
assert event.team == TeamType.HOME
event = next(events)
assert isinstance(event, SetupComplete)
event = next(events)
assert isinstance(event, Kickoff)
assert event.target == Position(1, 1)
assert event.scatter_direction == ScatterDirection.SW
assert event.scatter_distance == 1
assert board.get_ball_position() == Position(0, 0)
event = next(events)
assert isinstance(event, KickoffEventTuple)
assert event.result == KickoffEvent.CHEERING_FANS
event = next(events)
assert isinstance(event, Bounce)
assert event.scatter_direction == ScatterDirection.S
assert event.start_space == Position(0, 0)
assert event.end_space.is_offpitch()
event = next(events)
assert isinstance(event, Touchback)
assert event.player == player
assert not next(cmds, None)
assert not next(log_entries, None)
def test_touchback_for_own_half_kick_other_direction(board):
home_team, away_team = board.teams
replay = Replay(home_team, away_team, [], [])
player = home_team.get_player(0)
cmds = iter_([
SetupCompleteCommand(1, 0, TeamType.AWAY.value, 0, []),
SetupCompleteCommand(1, 0, TeamType.HOME.value, 0, []),
KickoffCommand(1, 0, TeamType.AWAY.value, 0, [8, 12]),
PreKickoffCompleteCommand(1, 0, TeamType.HOME.value, 0, [TeamType.HOME.value]),
TouchbackCommand(1, 0, TeamType.HOME.value, 0, [TeamType.HOME.value, 0])
])
log_entries = iter_([
KickDirectionLogEntry(TeamType.AWAY, "1", ScatterDirection.N.value),
KickDistanceLogEntry(TeamType.AWAY, "1", 1),
KickoffEventLogEntry(KickoffEvent.CHEERING_FANS.value),
BounceLogEntry(ScatterDirection.W.value)
])
events = replay._process_kickoff(cmds, log_entries, board)
event = next(events)
assert isinstance(event, TeamSetupComplete)
assert event.team == TeamType.AWAY
event = next(events)
assert isinstance(event, TeamSetupComplete)
assert event.team == TeamType.HOME
event = next(events)
assert isinstance(event, SetupComplete)
event = next(events)
assert isinstance(event, Kickoff)
assert event.target == Position(8, 12)
assert event.scatter_direction == ScatterDirection.N
assert event.scatter_distance == 1
assert board.get_ball_position() == Position(8, 13)
event = next(events)
assert isinstance(event, KickoffEventTuple)
assert event.result == KickoffEvent.CHEERING_FANS
event = next(events)
assert isinstance(event, Touchback)
assert event.player == player
assert not next(cmds, None)
assert not next(log_entries, None)