class TestSpiderCard(unittest.TestCase): #UT01 def setUp(self): self.card1 = SpiderCard('H', 3) #UT02 def test_getSuitNo(self): suitNo = self.card1.getSuitNo() self.assertEqual(suitNo, 2) #UT03 def test_getImage(self): deckGraphic = pygame.image.load('deck.png') pygame.init() pygame.display.set_mode((int(SpiderCard.card_size_x), SpiderCard.card_size_y)) self.testSurface = self.card1.getImage(deckGraphic) self.assertEqual(self.testSurface.get_width(), int(self.card1.card_size_x))
def setUp(self): self.card1 = SpiderCard('H', 3)