def testTroopCreateImageAssignsASurfaceToTroopImageAttribute(self): unit = Troop() unit.createImage() self.assertIs(type(unit.getImage()), pygame.Surface)
def testTroopGetImageForTroopWithANoneImageReturnsCorrectImage(self): unit = Troop() self.assertEqual(unit.getImage(), None)
def testTroopGetImageForManuallyAssignedImageReturnsCorrectImage(self): unit = Troop() image = pygame.Surface((10,10)) unit.image = image self.assertEqual(unit.getImage(), unit.image)
def testTroopInitializedWithCertainParametersHasCorrectValues(self): unit = Troop(5) self.assertEqual(unit.getId(), 5) self.assertEqual(unit.getImage(), None) self.assertEqual(unit.getRect(), None)