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)
