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)
