TorstenHettstedt/BattleTech
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

tests: nutze die zentrale Erstellung eines Mech

Browse Source
This commit is contained in:
TorstenHettstedt
2026-07-14 12:17:14 +02:00
parent 939b2cef92
commit 402370cf8a
3 changed files with 49 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_mech_bau.py
+34 -36
View File
@@ -1,48 +1,44 @@
import unittest
from hangar.mech import Mech
from mech_factory import MechFabrik
from mech_factory.component_storage import InnerStructure, AreaNames, Area, OuterAmor, Zone, ZoneNames
from mech_factory.component_storage.items import Cockpit, Gyroskope, Item, ItemTypes, Reaktor
from test_mech import AbstractTestMechAmor, INNER_ARMOR, OUTER_ARMOR
from test_mech import AbstractTestMech, AbstractTestMechAmor, INNER_ARMOR, OUTER_ARMOR
class TestMechBau(unittest.TestCase):
class TestMechBau(AbstractTestMech):
def test_simple(self):
fabrik = MechFabrik('Heuschrecke', 20.0, 8)
mech = fabrik.auslieferung()
self.assertIsInstance(mech, Mech)
self.assertIsInstance(mech.name, str)
self.assertEqual('Heuschrecke', mech.name)
self.assertIsInstance(mech.tonnage, float)
self.assertEqual(20.0, mech.tonnage)
self.assertIsInstance(mech.movement_rate, int)
self.assertEqual(8, mech.movement_rate)
self.mech = self.factory.auslieferung()
self.assertIsInstance(self.mech, Mech)
self.assertIsInstance(self.mech.name, str)
self.assertEqual('Heuschrecke', self.mech.name)
self.assertIsInstance(self.mech.tonnage, float)
self.assertEqual(20.0, self.mech.tonnage)
self.assertIsInstance(self.mech.movement_rate, int)
self.assertEqual(8, self.mech.movement_rate)
for ara_name in AreaNames:
area = mech.inner_structure.areas.get(ara_name)
area = self.mech.inner_structure.areas.get(ara_name)
should_amor = INNER_ARMOR[ara_name]
self.assertIsInstance(area, Zone)
self.assertIsInstance(area, Area)
self.assertEqual(should_amor, area.armor)
for zone_name in ZoneNames:
zone = mech.inner_structure.zones.get(zone_name)
zone = self.mech.inner_structure.zones.get(zone_name)
self.assertIsInstance(zone, Zone)
self.assertIsInstance(zone, OuterAmor)
self.assertEqual(0, zone.armor)
self.assertIsInstance(mech.inner_structure, InnerStructure)
self.assertIsInstance(mech.inner_structure.tonnage, float)
self.assertEqual(2.0, mech.inner_structure.tonnage)
self.assertIsInstance(self.mech.inner_structure, InnerStructure)
self.assertIsInstance(self.mech.inner_structure.tonnage, float)
self.assertEqual(2.0, self.mech.inner_structure.tonnage)
self.assertIsInstance(mech.inner_structure.gyroskope, Gyroskope)
self.assertIsInstance(mech.inner_structure.gyroskope.tonnage, float)
self.assertEqual(2.0, mech.inner_structure.gyroskope.tonnage)
self.assertIsInstance(mech.inner_structure.areas, dict)
self.assertIsInstance(self.mech.inner_structure.gyroskope, Gyroskope)
self.assertIsInstance(self.mech.inner_structure.gyroskope.tonnage, float)
self.assertEqual(2.0, self.mech.inner_structure.gyroskope.tonnage)
self.assertIsInstance(self.mech.inner_structure.areas, dict)
area_head = mech.inner_structure.areas.get(AreaNames.HEAD)
area_head = self.mech.inner_structure.areas.get(AreaNames.HEAD)
self.assertIsInstance(area_head, Area)
cockpit = area_head.cockpit
self.assertIsInstance(cockpit, Cockpit)
@@ -50,26 +46,26 @@ class TestMechBau(unittest.TestCase):
self.assertEqual(3.0, cockpit.tonnage)
self._test_for_head_items(area_head)
middle_torso = self._test_middle_torso(mech)
right_torso = self._test_areas(mech=mech, parent=middle_torso, index=AreaNames.RIGHT_TORSO, children=2)
left_torso = self._test_areas(mech=mech, parent=middle_torso, index=AreaNames.LEFT_TORSO, children=2)
middle_torso = self._test_middle_torso()
right_torso = self._test_areas(parent=middle_torso, index=AreaNames.RIGHT_TORSO, children=2)
left_torso = self._test_areas(parent=middle_torso, index=AreaNames.LEFT_TORSO, children=2)
self.assertIsInstance(area_head.parent, Area)
self.assertEqual(middle_torso, area_head.parent)
self.assertCountEqual([right_torso, left_torso], middle_torso.children)
right_arm = self._test_areas(mech=mech, parent=right_torso, index=AreaNames.RIGHT_ARM, children=0)
right_leg = self._test_areas(mech=mech, parent=right_torso, index=AreaNames.RIGHT_LEG, children=0)
right_arm = self._test_areas(parent=right_torso, index=AreaNames.RIGHT_ARM, children=0)
right_leg = self._test_areas(parent=right_torso, index=AreaNames.RIGHT_LEG, children=0)
self.assertCountEqual([right_arm, right_leg], right_torso.children)
self._test_for_arm_items(right_arm)
self._test_for_leg_items(right_leg)
left_arm = self._test_areas(mech=mech, parent=left_torso, index=AreaNames.LEFT_ARM, children=0)
left_leg = self._test_areas(mech=mech, parent=left_torso, index=AreaNames.LEFT_LEG, children=0)
left_arm = self._test_areas(parent=left_torso, index=AreaNames.LEFT_ARM, children=0)
left_leg = self._test_areas(parent=left_torso, index=AreaNames.LEFT_LEG, children=0)
self.assertCountEqual([left_arm, left_leg], left_torso.children)
self._test_for_arm_items(left_arm)
self._test_for_leg_items(left_leg)
def _test_middle_torso(self, mech) -> Area:
middle_torso = mech.inner_structure.areas.get(AreaNames.MIDDLE_TORSO)
def _test_middle_torso(self) -> Area:
middle_torso = self.mech.inner_structure.areas.get(AreaNames.MIDDLE_TORSO)
self.assertIsInstance(middle_torso, Area)
reactor = middle_torso.reactor
self.assertIsInstance(reactor, Reaktor)
@@ -81,8 +77,8 @@ class TestMechBau(unittest.TestCase):
self.assertIsInstance(middle_torso.children, list)
return middle_torso
def _test_areas(self, mech: Mech, parent: Area, index: AreaNames, children: int) -> Area:
area = mech.inner_structure.areas.get(index)
def _test_areas(self, parent: Area, index: AreaNames, children: int) -> Area:
area = self.mech.inner_structure.areas.get(index)
self.assertIsInstance(area, Area)
self.assertIsInstance(area.parent, Area)
self.assertEqual(parent, area.parent)
@@ -159,7 +155,9 @@ class TestMechBau(unittest.TestCase):
class TestMechAmor(AbstractTestMechAmor):
def test_build(self):
mech = self.factory.auslieferung()
for zone_name in ZoneNames:
zone = self.mech.inner_structure.zones.get(zone_name)
zone = mech.inner_structure.zones.get(zone_name)
should_outer_armor = OUTER_ARMOR[zone_name]
self.assertEqual(should_outer_armor, zone.armor)