예제 #1
0
    def test_get_color(self):
        c = Card(Card.SUIT_CLUBS, Card.JACK)
        color = c.get_color(c._suit)
        self.assertEqual(color, Card.COLOR_BLACK)

        c._suit = Card.SUIT_SPADES
        color = c.get_color(c._suit)
        self.assertEqual(color, Card.COLOR_BLACK)

        c._suit = Card.SUIT_DIAMONDS
        color = c.get_color(c._suit)
        self.assertEqual(color, Card.COLOR_RED)

        c._suit = Card.SUIT_HEARTS
        color = c.get_color(c._suit)
        self.assertEqual(color, Card.COLOR_RED)