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)
def testRegularCannonDoesNotEqualModifiedRegularCannon(self):
modified = types.new_attack('RegularCannon')
modified.power += 1
self.assertNotEqual(modified, types.new_attack('RegularCannon'))
modified = types.new_attack('RegularCannon')
modified.attackType = -1
self.assertNotEqual(modified, types.new_attack('RegularCannon'))
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 testRegularCannon(self):
regular_cannon = types.new_attack('RegularCannon')
self.assertEqual(regular_cannon.power, 5)
self.assertEqual(regular_cannon.attackType, types.attack_types['cannon'])
def testRegularCannonDoesNotEqualMachineGun(self):
self.assertNotEqual(types.new_attack('RegularCannon'), types.new_attack('MachineGun'))
def testRegularCannonEqualsRegularCannon(self):
self.assertEqual(types.new_attack('RegularCannon'), types.new_attack('RegularCannon'))
def testDoubleMachineGun(self):
double_machine_gun = types.new_attack('DoubleMachineGun')
self.assertEqual(double_machine_gun.power, 10)
self.assertEqual(double_machine_gun.attackType, types.attack_types['bullet'])
def testMachineGun(self):
machine_gun = types.new_attack('MachineGun')
self.assertEqual(machine_gun.power, 5)
self.assertEqual(machine_gun.attackType, types.attack_types['bullet'])
