def test_bracket_results(self):
"""Test the results of playing losers bracket."""
teams = ['Team {}'.format(x + 1) for x in range(7)]
bracket = PowerOfTwoLosersBracket(rounds=3)
for team in teams:
bracket.add_team(team)
for i in range(5):
self.assertIsInstance(bracket.play_match(), unicode, 'Cup ended early.')
self.assertRaises(CupFinished, bracket.play_match)
final_match = bracket.matches[3][0]
self.assertEqual(final_match.team1, teams[6], 'First finalist is wrong.')
self.assertEqual(final_match.team2, teams[5], 'Second finalist is wrong.')
self.assertEqual(bracket.winner, 'Team 7', 'Cup has wrong winner.')
def __init__(self, rounds=0, teams=[], require_double_win=True):
"""Constructor.
@param rounds: The number of rounds
@type rounds: int
@param teams: An optional list of teams
@type teams: list
"""
# Parent constructor
super(PowerOfTwoDoubleEliminationCup, self).__init__(teams=teams,
team_count=2 ** rounds)
# Parameters
self.require_double_win = require_double_win
Match = config.base_match # Load match class
# Bracket setup
self.winners_bracket = PowerOfTwoSingleEliminationCup(rounds=rounds,
teams=self.teams)
self.losers_bracket = PowerOfTwoLosersBracket(rounds=rounds)
self.bracket_progression = [self.losers_bracket, self.winners_bracket,
self.losers_bracket] * (rounds - 1)
self.current_bracket = self.winners_bracket
self.current_bracket_index = -1
self.final = Match('Winners Bracket Winner', 'Losers Bracket Winner')
self.final_scores = [('', ''), ('', '')]
self.winners = {}
class PowerOfTwoDoubleEliminationCup(StandardCup):
"""Standard double-elimination cup for powers of two (4, 8, 16, etc.)."""
def __init__(self, rounds=0, teams=[], require_double_win=True):
"""Constructor.
@param rounds: The number of rounds
@type rounds: int
@param teams: An optional list of teams
@type teams: list
"""
# Parent constructor
super(PowerOfTwoDoubleEliminationCup, self).__init__(teams=teams,
team_count=2 ** rounds)
# Parameters
self.require_double_win = require_double_win
Match = config.base_match # Load match class
# Bracket setup
self.winners_bracket = PowerOfTwoSingleEliminationCup(rounds=rounds,
teams=self.teams)
self.losers_bracket = PowerOfTwoLosersBracket(rounds=rounds)
self.bracket_progression = [self.losers_bracket, self.winners_bracket,
self.losers_bracket] * (rounds - 1)
self.current_bracket = self.winners_bracket
self.current_bracket_index = -1
self.final = Match('Winners Bracket Winner', 'Losers Bracket Winner')
self.final_scores = [('', ''), ('', '')]
self.winners = {}
def _play_match(self):
"""Play a match. Internal code."""
if self.current_bracket.round_over:
self.current_bracket_index += 1
self.current_bracket = self.bracket_progression[self.current_bracket_index]
winner = self.current_bracket.play_match()
if self.current_bracket == self.winners_bracket:
loser = self.current_bracket.current_match.loser
self.losers_bracket.add_team(loser)
return winner
def _bracket_finished(self, cf):
"""Handle a finished sub-bracket."""
self.winners[self.current_bracket] = cf.winner
if self.current_bracket == self.winners_bracket:
self.final.team1 = cf.winner
loser = self.current_bracket.current_match.loser
self.losers_bracket.add_team(loser)
else:
self.final.team2 = cf.winner
return cf.winner
def _play_final(self):
"""Play the cup final."""
winner = self.final.play()
self.final_scores[0] = (self.final.score1, self.final.score2)
if not self.require_double_win or winner == self.winners[self.winners_bracket]:
self.winner = winner
raise CupFinished(winner)
else:
self.winner = self.final.play()
self.final_scores[1] = (self.final.score1, self.final.score2)
raise CupFinished(self.winner)
def play_match(self):
"""Play a cup match.
@return: The winner of the simulated match
@raise CupFinished: If the cup is finished
"""
try:
return self._play_match()
except CupFinished as cf:
return self._bracket_finished(cf)
except IndexError:
self._play_final()
def update_teams(self, teams):
"""Update the list of teams and the first-round matches.
@param teams: The new list of teams
@type teams: list
"""
self.teams = teams
self.winners_bracket.update_teams(teams)
def print_cup(self, display=True):
"""Print the cup to a string and (optionally) the console.
@param display: Whether to print to the console.
@type display: bool
@return: The displayed bracket
@rtype: str
"""
winners_printout = self.winners_bracket.print_cup(display=False)
losers_printout = self.losers_bracket.print_cup(display=False)
final_printout = (
'\n'
'{:<30} {:>4} {:>4} '
' '
'\n'
' '
' '
'\n'
'{:<30} {:>4} {:>4} '
' '
'\n'
).format(self.final.team1, self.final_scores[0][0], self.final_scores[1][0],
self.final.team2, self.final_scores[0][1], self.final_scores[1][1])
full_printout = winners_printout + losers_printout + '\n' + final_printout
if display:
print(full_printout)
return full_printout
def test_bracket_printout(self):
"""Test the printout of the losers bracket when completed."""
teams = ['Team {}'.format(x + 1) for x in range(7)]
bracket = PowerOfTwoLosersBracket(rounds=3)
for team in teams:
bracket.add_team(team)
try:
while True:
bracket.play_match()
except CupFinished:
pass
expected_string = (
'\n'
' '
' '
' '
'Team 7 5 '
'\n'
' '
'Team 6 5 '
' '
' '
'\n'
'Team 1 5 '
' '
'Team 6 5 '
' '
'\n'
' '
'Team 1 0 '
' '
' '
'\n'
'Team 2 0 '
' '
' '
'Team 6 0 '
'\n'
' '
'Team 5 5 '
' '
' '
'\n'
'Team 3 5 '
' '
'Team 5 0 '
' '
'\n'
' '
'Team 3 0 '
' '
' '
'\n'
'Team 4 0 '
' '
' '
' '
'\n'
)
self.assertEqual(bracket.print_cup(), expected_string,
'Wrong bracket printed.')
