diff --git a/tests/test_mech_bau.py b/tests/test_mech_bau.py index b93849a..20ef4f3 100755 --- a/tests/test_mech_bau.py +++ b/tests/test_mech_bau.py @@ -188,8 +188,8 @@ class TextMechAmor(unittest.TestCase): def test_process(self): for area_name in AreaNames: - self.mech.strike(area_name, 3) - 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.mech.strike(area_name, OUTER_ARMOR[area_name] - 3) + self.assertEqual(3, self.mech.inner_structure.areas.get(area_name).armor) + self.mech.strike(area_name, 4) 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) + self.assertEqual(INNER_ARMOR[area_name] - 1, self.mech.inner_structure.areas.get(area_name).inner_armor)