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)