def testUpgradeTank(self): map = Game(1, 13) tank = Tank(1, [2, 3], map) bonus = VestBonus(0, [2, 2]) self.assertEqual(tank.getMaxBullets(), 1) bonus.upgradeTank(tank) self.assertEqual(tank.getMaxBullets(), 2) bonus.downgradeTank(tank) self.assertEqual(tank.getMaxBullets(), 1)
def testMaxBullets(self): map = Game(1, 13) tank = Tank(1, [1, 2], map) self.assertEqual(tank.getMaxBullets(), 1) tank.setMaxBullets(2) self.assertEqual(tank.getMaxBullets(), 2)