def testRecon(self):
recon = types.new_unit('Recon', consts.RED)
self.assertEqual(recon.health, 10)
self.assertEqual(recon.attacks, [types.new_attack('DoubleMachineGun')])
self.assertEqual(recon.armor, types.new_armor('WeakMetal'))
self.assertEqual(recon.movement, types.new_movement('Tires'))
self.assertEqual(recon.distance, 9)
def testInfantry(self):
inf = types.new_unit('Infantry', consts.RED)
self.assertEqual(inf.health, 10)
self.assertEqual(inf.attacks, [types.new_attack('MachineGun')])
self.assertEqual(inf.armor, types.new_armor('BodyArmor'))
self.assertEqual(inf.movement, types.new_movement('Feet'))
self.assertEqual(inf.distance, 3)
def testTank(self):
tank = types.new_unit('Tank', consts.RED)
self.assertEqual(tank.health, 10)
self.assertEqual(
tank.attacks, [types.new_attack('RegularCannon'), types.new_attack('MachineGun')])
self.assertEqual(tank.armor, types.new_armor('HeavyMetal'))
self.assertEqual(tank.movement, types.new_movement('Treads'))
self.assertEqual(tank.distance, 7)
