Exemplo n.º 1
0
class DeckTests(unittest.TestCase):
    def setUp(self):
        self.cardlist = [Card('One', 1),
                         Card('Two', 2)]
        self.deck = Deck(self.cardlist)

    def testDraw(self):
        """Deck draw"""
        card = self.deck.draw()
        self.assertTrue(card in self.cardlist)
        self.assertEqual(1, len(self.deck.drawn))
        self.assertEqual(1, len(self.deck.undrawn))

    def testEmptyDraw(self):
        """Deck empty draw"""
        self.deck.draw()
        self.deck.draw()
        self.assertEqual(None, self.deck.draw())

    def testShuffle(self):
        """Deck shuffle"""
        self.deck.draw()
        self.deck.shuffle()
        self.assertEqual(2, len(self.deck.undrawn))
        self.assertEqual(0, len(self.deck.drawn))
Exemplo n.º 2
0
 def setUp(self):
     self.cardlist = [Card('One', 1),
                      Card('Two', 2)]
     self.deck = Deck(self.cardlist)