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)