def testTroopGetRectForManuallyAssignedRectReturnsCorrectRect(self): unit = Troop() surface = pygame.Surface((10,10)) rect = surface.get_rect() unit.rect = rect self.assertEqual(unit.getRect(), unit.rect)
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 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 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)
def testTroopGetImageForManuallyAssignedImageReturnsCorrectImage(self): unit = Troop() image = pygame.Surface((10,10)) unit.image = image self.assertEqual(unit.getImage(), unit.image)
def testTroopGetRectForTroopWithANoneRectReturnsCorrectRect(self): unit = Troop() self.assertEqual(unit.getRect(), None)
def testTroopGetImageForTroopWithANoneImageReturnsCorrectImage(self): unit = Troop() self.assertEqual(unit.getImage(), None)
def testTroopGetIdWithManuallyAssignedIdReturnsCorrectId(self): unit = Troop() unit._id = 83 self.assertEqual(unit.getId(), 83)
def testTroopGetIdWithUnassignedIdReturnsCorrectId(self): unit = Troop() self.assertEqual(unit.getId(), -1)
def testTroopInitializedWithCertainParametersHasCorrectValues(self): unit = Troop(5) self.assertEqual(unit.getId(), 5) self.assertEqual(unit.getImage(), None) self.assertEqual(unit.getRect(), None)
def testTroopInitializedWithNoParametersHasCorrectValues(self): unit = Troop() self.assertEqual(unit.getId(), -1) self.assertEqual(unit.image, None) self.assertEqual(unit.rect, None)