class test_hanabi(unittest2.TestCase):
def setUpGame(self):
self.game = Game()
for p in players:
self.game.add_player(p)
self.game.start_game(players[0])
self.game.turn_order = list(players)
def getHand(self, h):
return ' '.join([str(c) for c in h])
def getBacks(self, h):
return ''.join([c.back() for c in h])
def getFronts(self, h):
return ''.join([c.front() for c in h])
def test_handmgt(self):
p = Player(players[0])
p.hand = [Card('red', i, b) for i, b in zip(xrange(1,6), uppercase[:5])]
self.assertEqual('ABCDE', self.getBacks(p.hand))
print p.swap_cards('A', 'E')
self.assertEqual('EBCDA', self.getBacks(p.hand))
def test_play(self):
self.setUpGame()
print self.game.turn()
print self.game.play_card(players[0], 'A')
print self.game.turn()
print self.game.hint_player(players[1], players[0], 'blue')
class test_hanabi(unittest2.TestCase):
def setUpGame(self):
self.game = Game()
self.game.markup = xterm_markup()
for p in players:
self.game.add_player(p)
for c in self.game.deck:
c.markup = xterm_markup()
self.game.start_game(players[0])
self.game.turn_order = list(players)
def getHand(self, h):
return ' '.join([str(c) for c in h])
def getBacks(self, h):
return ''.join([c.back() for c in h])
def getFronts(self, h):
return ''.join([c.front() for c in h])
def test_handmgt(self):
p = Player(players[0])
p.hand = [Card('red', i, b) for i, b in zip(xrange(1,6), uppercase[:5])]
for c in p.hand:
c.markup = xterm_markup()
self.assertEqual('ABCDE', self.getBacks(p.hand))
print p.swap_cards('A', 'E')
self.assertEqual('EBCDA', self.getBacks(p.hand))
def test_play(self):
self.setUpGame()
print self.game.turn()
print self.game.play_card(players[0], 'A')
print self.game.turn()
print self.game.hint_player(players[1], players[0], 'blue')
def test_show_hints(self):
p0, p1 = players[0], players[1]
self.setUpGame()
self.game.hint_player(p0, p1, 1)
self.game.hint_player(p1, p0, 1)
self.game.hint_player(p0, p1, 1)
print self.game.hints(p0)
hints = self.game.hints(p0)
self.assertTrue(len(hints.private[p0]) == 1)
print self.game.hints(p1, show_all=True)
hints = self.game.hints(p1, show_all=True)
self.assertTrue(len(hints.private[p1]) == 3)
def test_lastround(self):
# Make the deck have one card, let a player discard.
# Then make sure each player (inc. initial one) gets one
# more turn.
self.setUpGame()
self.game.deck = [Card('red', 1, 'A')]
self.game.options['repeat_backs']['value'] = True
for c in self.game.deck:
c.markup = xterm_markup()
self.game.discard_card(self.game.player_turn(), 'A')
for i in xrange(len(players)):
self.assertFalse(self.game.game_over())
print self.game.discard_card(self.game.player_turn(), 'A')
self.assertTrue(self.game.game_over())
def test_rainbow_display(self):
p = Player(players[0])
p.hand = [Card('rainbow', i, b) for i, b in zip(xrange(1,6), uppercase[:5])]
m = xterm_markup()
print m.color('hello world', xterm_markup.RAINBOW)
self.game = Game()
self.game.markup = m
for p in players:
self.game.add_player(p)
for c in self.game.deck:
c.markup = xterm_markup()
self.game.start_game(players[0], opts={'rainbow_10': True})
self.game.turn_order = list(players)
for n, l in [(1, 'A'), (2, 'B'), (3, 'C')]:
c = Card(text_markup_base.RAINBOW, n, l)
c.markup = self.game.markup
self.game._players[self.game.player_turn()].hand[n-1] = c
self.game.play_card(self.game.player_turn(), 'A')
self.game.play_card(self.game.player_turn(), 'A')
print self.game.play_card(self.game.player_turn(), 'B')
def test_hints(self):
self.setUpGame()
p1, p2 = self.game.turn_order[0], self.game.turn_order[1]
self.game.hint_player(p1, p2, 5)
self.game.hint_player(p2, p1, 5)
self.game.hint_player(p1, p2, 5)
self.game.hint_player(p2, p1, 5)
self.game.hint_player(p1, p2, 5)
gr = self.game.hints(p1)
self.assertTrue(len(gr.private[p1]) == 2)
gr = self.game.hints(p2)
self.assertTrue(len(gr.private[p2]) == 3)
def test_watch(self):
self.setUpGame()
p1 = self.game.turn_order[0]
print self.game.add_watcher('henry')
self.assertTrue('henry' in self.game._watchers)
print self.game.discard_card(p1, 'A')
print self.game.remove_player('henry')
self.assertFalse('henry' in self.game._watchers)
print self.game.add_watcher(p1)
self.assertTrue(not p1 in self.game._watchers)
def test_unsolvable_rainbow_5(self):
game = Game()
game.markup = xterm_markup()
for p in players:
game.add_player(p)
for c in game.deck:
c.markup = xterm_markup()
opts = {'rainbow_5': True}
game.options['solvable_rainbow_5'] = True
bad_card = Card('rainbow', 1)
bad_card.markup = xterm_markup()
game.deck[len(game.deck)-1] = bad_card
last_card = game.deck[len(game.deck)-1]
print '\nlast card before: %s' % last_card
game.start_game(players[0], opts)
last_card = game.deck[len(game.deck)-1]
print 'last card after: %s' % last_card
self.assertFalse(last_card.color == 'rainbow' and
last_card.number in [1,2,3,4])
