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)