diff --git a/tests/test_mech_bau.py b/tests/test_mech_bau.py index fe53b89..b93849a 100755 --- a/tests/test_mech_bau.py +++ b/tests/test_mech_bau.py @@ -3,7 +3,7 @@ import unittest from MechFabrik import MechFabrik from MechFabrik.bauteile import InnerStructure, Gyroskope, AreaNames, Area, Cockpit, Item, ItemTypes, Reaktor, Mech -INNER_AMOR = { +INNER_ARMOR = { AreaNames.HEAD: 3, AreaNames.MIDDLE_TORSO: 6, AreaNames.LEFT_TORSO: 5, @@ -14,7 +14,7 @@ INNER_AMOR = { AreaNames.RIGHT_LEG: 4, } -OUTER_AMOR = { +OUTER_ARMOR = { AreaNames.HEAD: 6, AreaNames.MIDDLE_TORSO: 9, AreaNames.LEFT_TORSO: 8, @@ -41,7 +41,7 @@ class TestMechBau(unittest.TestCase): for ara_name in AreaNames: area = mech.inner_structure.areas.get(ara_name) - should_inner_amor = INNER_AMOR[ara_name] + should_inner_amor = INNER_ARMOR[ara_name] self.assertEqual(should_inner_amor, area.inner_armor) self.assertEqual(0, area.armor) @@ -171,11 +171,11 @@ class TextMechAmor(unittest.TestCase): def setUp(self): fabrik = MechFabrik('Heuschrecke', 20.0, 8) - fabrik.setHeadAmore(OUTER_AMOR[AreaNames.HEAD]) - fabrik.setMidleTorsoAmore(OUTER_AMOR[AreaNames.MIDDLE_TORSO]) - fabrik.setSideTorsoAmore(OUTER_AMOR[AreaNames.LEFT_TORSO]) - fabrik.setArmAmore(OUTER_AMOR[AreaNames.LEFT_ARM]) - fabrik.setLegAmore(OUTER_AMOR[AreaNames.LEFT_LEG]) + fabrik.set_head_armor(OUTER_ARMOR[AreaNames.HEAD]) + fabrik.set_middle_torso_armor(OUTER_ARMOR[AreaNames.MIDDLE_TORSO]) + fabrik.set_side_torso_armor(OUTER_ARMOR[AreaNames.LEFT_TORSO]) + fabrik.set_arm_armor(OUTER_ARMOR[AreaNames.LEFT_ARM]) + fabrik.set_leg_armor(OUTER_ARMOR[AreaNames.LEFT_LEG]) self.mech = fabrik.auslieferung() @@ -183,13 +183,13 @@ class TextMechAmor(unittest.TestCase): for area_name in AreaNames: area = self.mech.inner_structure.areas.get(area_name) - should_outer_amor = OUTER_AMOR[area_name] - self.assertEqual(should_outer_amor, area.armor) + should_outer_armor = OUTER_ARMOR[area_name] + self.assertEqual(should_outer_armor, area.armor) def test_process(self): for area_name in AreaNames: self.mech.strike(area_name, 3) - self.assertEqual(OUTER_AMOR[area_name] - 3, self.mech.inner_structure.areas.get(area_name).armor) - self.mech.strike(area_name, OUTER_AMOR[area_name] - 2) + self.assertEqual(OUTER_ARMOR[area_name] - 3, self.mech.inner_structure.areas.get(area_name).armor) + self.mech.strike(area_name, OUTER_ARMOR[area_name] - 2) self.assertEqual(0, self.mech.inner_structure.areas.get(area_name).armor) self.assertEqual(2, self.mech.inner_structure.areas.get(area_name).inner_armor)