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_split_turnovers(self):
self.__set_turnover_consts()
# Successful
self.assertEqual(split_turnovers(self.turnovers[0]),
self.turnover_splits[0])
self.assertEqual(split_turnovers(self.turnovers[1]),
self.turnover_splits[1])
self.assertEqual(split_turnovers(self.turnovers[2]),
self.turnover_splits[2])
self.assertEqual(split_turnovers(self.turnovers[3]),
self.turnover_splits[3])
self.assertEqual(split_turnovers(self.turnovers[4]),
self.turnover_splits[4])
self.assertEqual(split_turnovers(self.turnovers[5]),
self.turnover_splits[5])
self.assertEqual(split_turnovers(self.turnovers[6]),
self.turnover_splits[6])
self.assertEqual(split_turnovers(self.turnovers[7]),
self.turnover_splits[7])
self.assertEqual(split_turnovers(self.turnovers[8]),
self.turnover_splits[8])
self.assertEqual(split_turnovers(self.turnovers[9]),
self.turnover_splits[9])
self.assertEqual(split_turnovers(self.turnovers[10]),
self.turnover_splits[10])
self.assertEqual(split_turnovers(self.turnovers[11]),
self.turnover_splits[11])
self.assertEqual(split_turnovers(self.turnovers[12]),
self.turnover_splits[12])
self.assertEqual(split_turnovers(self.turnovers[13]),
self.turnover_splits[13])
def test_split_turnovers(self):
self.__set_turnover_consts()
# Successful
self.assertEqual(
split_turnovers(self.turnovers[0]),
self.turnover_splits[0]
)
self.assertEqual(
split_turnovers(self.turnovers[1]),
self.turnover_splits[1]
)
self.assertEqual(
split_turnovers(self.turnovers[2]),
self.turnover_splits[2]
)
self.assertEqual(
split_turnovers(self.turnovers[3]),
self.turnover_splits[3]
)
self.assertEqual(
split_turnovers(self.turnovers[4]),
self.turnover_splits[4]
)
self.assertEqual(
split_turnovers(self.turnovers[5]),
self.turnover_splits[5]
)
self.assertEqual(
split_turnovers(self.turnovers[6]),
self.turnover_splits[6]
)
self.assertEqual(
split_turnovers(self.turnovers[7]),
self.turnover_splits[7]
)
self.assertEqual(
split_turnovers(self.turnovers[8]),
self.turnover_splits[8]
)
self.assertEqual(
split_turnovers(self.turnovers[9]),
self.turnover_splits[9]
)
self.assertEqual(
split_turnovers(self.turnovers[10]),
self.turnover_splits[10]
)
self.assertEqual(
split_turnovers(self.turnovers[11]),
self.turnover_splits[11]
)
self.assertEqual(
split_turnovers(self.turnovers[12]),
self.turnover_splits[12]
)
self.assertEqual(
split_turnovers(self.turnovers[13]),
self.turnover_splits[13]
)
