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 testTroopCreateImageAssignsARectToTroopRectAttribute(self):
unit = Troop()
unit.createImage()
self.assertIs(type(unit.getRect()), pygame.Rect)
def testTroopGetRectForTroopWithANoneRectReturnsCorrectRect(self):
unit = Troop()
self.assertEqual(unit.getRect(), None)
def testTroopInitializedWithCertainParametersHasCorrectValues(self):
unit = Troop(5)
self.assertEqual(unit.getId(), 5)
self.assertEqual(unit.getImage(), None)
self.assertEqual(unit.getRect(), None)
