Esempio n. 1
0
    def testTroopCreateImageAssignsASurfaceToTroopImageAttribute(self):
        unit = Troop()
        unit.createImage()

        self.assertIs(type(unit.getImage()), pygame.Surface)
Esempio n. 2
0
    def testTroopGetImageForTroopWithANoneImageReturnsCorrectImage(self):
        unit = Troop()

        self.assertEqual(unit.getImage(), None)
Esempio n. 3
0
    def testTroopGetImageForManuallyAssignedImageReturnsCorrectImage(self):
        unit = Troop()
        image = pygame.Surface((10,10))
        unit.image = image

        self.assertEqual(unit.getImage(), unit.image)
Esempio n. 4
0
    def testTroopInitializedWithCertainParametersHasCorrectValues(self):
        unit = Troop(5)

        self.assertEqual(unit.getId(), 5)
        self.assertEqual(unit.getImage(), None)
        self.assertEqual(unit.getRect(), None)