class TournamentTest(unittest.TestCase): def setUp(self): self.player1 = ComputerPlayer('X') self.player2 = ComputerPlayer('O') self.players = (self.player1, self.player2) self.tournament = Tournament(self.players) def testPlayedGameEnds(self): winner = None while not winner: winner = self.tournament.play_game() self.assertEquals(1, winner.games_won) def testGetWinnerOrNone(self): self.assertIs(self.tournament._get_winner_or_none(None), None) self.assertIs(self.tournament._get_winner_or_none('O'), self.player2)