def testTroopColorSpriteMethodCallsSurfaceFillMethod(self):
unit = Troop()
unit.createImage()
unit.image = Mock()
unit.colorSprite((0,0,0))
unit.image.fill.assert_called_once_with((0,0,0))
def testTroopRectAttributeHasCorrectDimensionsAfterCreateImageCall(self):
unit = Troop()
unit.createImage()
unit_rect = unit.getRect()
correct_width = 10
correct_height = 10
self.assertEqual(unit_rect.width, correct_width)
self.assertEqual(unit_rect.height, correct_height)
def testTroopCreateImageAssignsARectToTroopRectAttribute(self):
unit = Troop()
unit.createImage()
self.assertIs(type(unit.getRect()), pygame.Rect)
def testTroopCreateImageAssignsASurfaceToTroopImageAttribute(self):
unit = Troop()
unit.createImage()
self.assertIs(type(unit.getImage()), pygame.Surface)
