def test_fumble_then_prone_bounce(board):
home_team, away_team = board.teams
player = home_team.get_player(0)
board.set_position(Position(13, 11), player)
opponent = away_team.get_player(0)
board.set_position(Position(14, 10), opponent)
board.set_prone(opponent)
replay = Replay(home_team, away_team, [], [])
board.set_ball_position(Position(13, 12))
board.setup_complete()
cmds = iter_([
DeclineRerollCommand(1, 1, TeamType.HOME, 1, [])
])
log_entries = [
BounceLogEntry(ScatterDirection.S.value),
CatchEntry(TeamType.HOME, 1, "3+", "1", ActionResult.FAILURE.name),
BounceLogEntry(ScatterDirection.SE.value),
BounceLogEntry(ScatterDirection.W.value)
]
log_entries_iter = iter_(log_entries)
events = replay._process_ball_movement(cmds, log_entries_iter, board)
event = next(events)
assert isinstance(event, Bounce)
assert event.start_space == Position(13, 12)
assert event.end_space == Position(13, 11)
event = next(events)
assert isinstance(event, Action)
assert event.player == player
assert event.action == ActionType.CATCH
assert event.result == ActionResult.FAILURE
event = next(events)
assert isinstance(event, Bounce)
assert event.start_space == Position(13, 11)
assert event.end_space == Position(14, 10)
event = next(events)
assert isinstance(event, Bounce)
assert event.start_space == Position(14, 10)
assert event.end_space == Position(13, 10)
assert not next(cmds, None)
assert not next(log_entries_iter, None)
assert not next(events, None)
def test_ghost_bounce_due_to_no_catch(board):
# Based on turn 6 of Replay_2021-04-04_09-49-09.db when Kardel the Putrefier is knocked into the ball
home_team, away_team = board.teams
player = home_team.get_player(0)
board.set_position(Position(13, 11), player)
replay = Replay(home_team, away_team, [], [])
board.set_ball_position(Position(13, 12))
board.setup_complete()
cmds = iter_([])
log_entries = [
BounceLogEntry(ScatterDirection.S.value),
BounceLogEntry(ScatterDirection.SE.value)
]
log_entries_iter = iter_(log_entries)
events = replay._process_ball_movement(cmds, log_entries_iter, board)
event = next(events)
assert isinstance(event, Bounce)
assert event.start_space == Position(13, 12)
assert event.end_space == Position(14, 11)
assert not next(events, None)
def test_throwin_direction_across_field(board):
home_team, away_team = board.teams
player = home_team.get_player(0)
board.set_position(Position(7, 12), player)
replay = Replay(home_team, away_team, [], [])
board.set_ball_position(Position(0, 7))
board.setup_complete()
cmds = [
]
log_entries = [
BounceLogEntry(ScatterDirection.W.value),
ThrowInDirectionLogEntry(ThrowInDirection.CENTRE.value),
ThrowInDistanceLogEntry(3),
BounceLogEntry(ScatterDirection.W.value),
TurnOverEntry(TeamType.AWAY, "Fumble")
]
cmds_iter = iter_(cmds)
log_entries_iter = iter_(log_entries)
events = replay._process_ball_movement(cmds, log_entries_iter, board)
event = next(events)
assert isinstance(event, Bounce)
assert event.start_space == Position(0, 7)
assert event.end_space == Position(-1, 7)
event = next(events)
assert isinstance(event, ThrowIn)
assert event.start_space == Position(0, 7)
assert event.end_space == Position(3, 7)
event = next(events)
assert isinstance(event, Bounce)
assert event.start_space == Position(3, 7)
assert event.end_space == Position(2, 7)
assert not next(events, None)
assert not next(cmds_iter, None)
assert isinstance(next(log_entries_iter), TurnOverEntry)
print(f"Found replay file with no matching log - {db_path}", sys.stderr)
continue
# Duplicate the `create_replay()` function because we need to wrap the commands
db = sqlite3.connect(db_path)
home_team = create_team(db, TeamType.HOME)
away_team = create_team(db, TeamType.AWAY)
commands = create_commands(db)
log_entries = parse_log_entries(log_path)
num_commands = len(commands)
total_commands += num_commands
# Wrap commands in an iter so we can track how far the process got
commands = iter(commands)
replay = Replay(home_team, away_team, commands, log_entries)
i = 0
try:
replay.validate()
# We can't just do len() because it might throw an exception
# But we don't care about the content, just the number
for _ in replay.events():
i += 1
except: # noqa: E722 - we explicitly don't want to stop on anything
pass
finally:
num_commands_unprocessed = len(list(commands))
num_commands_processed = num_commands - num_commands_unprocessed