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'])