def __set_turnover(self):
""" Takes the description of a play from self with a turnover in it and
sets the "turnover" dictionary for the play.
returns:
A turnover list with the following fields:
"turnovers": [{ "type": "fumble", "by": "home", "recovered":
"away" }, ...]
"""
turnovers = []
turnover = {}
for turn_string in split_turnovers(self.current_play_info["description"]):
t = deepcopy(turnover)
# Set the name of the penalty
t["type"] = get_turnover_type(turn_string)
# Set the teams
(com, rec) = get_turnover_teams(
turn_string,
self.home_players,
self.away_players
)
if com:
t["by"] = com
if rec:
t["recovered"] = rec
# Fill in the full dictionary
turnovers.append(t)
return turnovers
def test_get_turnover_type(self):
self.__set_turnover_consts()
# Successful
self.assertEqual(get_turnover_type(self.turnover_splits[0][0]),
"fumble")
self.assertEqual(get_turnover_type(self.turnover_splits[1][0]),
"interception")
self.assertEqual(get_turnover_type(self.turnover_splits[2][0]),
"interception")
self.assertEqual(get_turnover_type(self.turnover_splits[3][0]),
"fumble")
self.assertEqual(get_turnover_type(self.turnover_splits[4][0]),
"fumble")
self.assertEqual(get_turnover_type(self.turnover_splits[5][0]),
"interception")
self.assertEqual(get_turnover_type(self.turnover_splits[5][1]),
"fumble")
self.assertEqual(get_turnover_type(self.turnover_splits[6][0]),
"fumble")
self.assertEqual(get_turnover_type(self.turnover_splits[7][0]),
"fumble")
self.assertEqual(get_turnover_type(self.turnover_splits[8][0]),
"fumble")
self.assertEqual(get_turnover_type(self.turnover_splits[9][0]),
"muffed catch")
self.assertEqual(get_turnover_type(self.turnover_splits[10][0]),
"interception")
self.assertEqual(get_turnover_type(self.turnover_splits[10][1]),
"fumble")
self.assertEqual(get_turnover_type(self.turnover_splits[11][0]),
"fumble")
self.assertEqual(get_turnover_type(self.turnover_splits[12][0]),
"muffed catch")
def test_get_turnover_type(self):
self.__set_turnover_consts()
# Successful
self.assertEqual(
get_turnover_type(self.turnover_splits[0][0]),
"fumble"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[1][0]),
"interception"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[2][0]),
"interception"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[3][0]),
"fumble"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[4][0]),
"fumble"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[5][0]),
"interception"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[5][1]),
"fumble"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[6][0]),
"fumble"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[7][0]),
"fumble"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[8][0]),
"fumble"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[9][0]),
"muffed catch"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[10][0]),
"interception"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[10][1]),
"fumble"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[11][0]),
"fumble"
)
self.assertEqual(
get_turnover_type(self.turnover_splits[12][0]),
"muffed catch"
)
