Files
BattleTech/tests/test_mech/__init__.py
T

61 lines
2.3 KiB
Python

import unittest
from mech_factory import MechFabrik
from mech_factory.component_storage import AreaNames, ZoneNames
INNER_ARMOR = {
AreaNames.HEAD: 3,
AreaNames.MIDDLE_TORSO: 6,
AreaNames.LEFT_TORSO: 5,
AreaNames.RIGHT_TORSO: 5,
AreaNames.LEFT_ARM: 3,
AreaNames.RIGHT_ARM: 3,
AreaNames.LEFT_LEG: 4,
AreaNames.RIGHT_LEG: 4,
}
## entspricht 64 Punkte → 4 t Panzerung
OUTER_ARMOR = {
ZoneNames.HEAD: 6,
ZoneNames.MIDDLE_TORSO_FRONT: 8,
ZoneNames.MIDDLE_TORSO_REAR: 4,
ZoneNames.LEFT_TORSO_FRONT: 8,
ZoneNames.LEFT_TORSO_REAR: 3,
ZoneNames.RIGHT_TORSO_FRONT: 8,
ZoneNames.RIGHT_TORSO_REAR: 3,
ZoneNames.LEFT_ARM: 5,
ZoneNames.RIGHT_ARM: 5,
ZoneNames.LEFT_LEG: 7,
ZoneNames.RIGHT_LEG: 7,
}
NEXT_OUTER_ZONE = {
ZoneNames.HEAD: AreaNames.HEAD,
ZoneNames.MIDDLE_TORSO_FRONT: AreaNames.MIDDLE_TORSO,
ZoneNames.MIDDLE_TORSO_REAR: AreaNames.MIDDLE_TORSO,
ZoneNames.LEFT_TORSO_FRONT: AreaNames.LEFT_TORSO,
ZoneNames.LEFT_TORSO_REAR: AreaNames.LEFT_TORSO,
ZoneNames.RIGHT_TORSO_FRONT: AreaNames.RIGHT_TORSO,
ZoneNames.RIGHT_TORSO_REAR: AreaNames.RIGHT_TORSO,
ZoneNames.LEFT_ARM: AreaNames.LEFT_ARM,
ZoneNames.RIGHT_ARM: AreaNames.RIGHT_ARM,
ZoneNames.LEFT_LEG: AreaNames.LEFT_LEG,
ZoneNames.RIGHT_LEG: AreaNames.RIGHT_LEG,
}
class AbstractTestMech(unittest.TestCase):
def setUp(self):
self.factory = MechFabrik('Heuschrecke', 20.0, 8)
class AbstractTestMechAmor(AbstractTestMech):
def setUp(self):
super(AbstractTestMechAmor, self).setUp()
self.factory.set_head_armor(OUTER_ARMOR[ZoneNames.HEAD])
self.factory.set_middle_torso_armor(OUTER_ARMOR[ZoneNames.MIDDLE_TORSO_FRONT],OUTER_ARMOR[ZoneNames.MIDDLE_TORSO_REAR])
self.factory.set_side_torso_armor(OUTER_ARMOR[ZoneNames.LEFT_TORSO_FRONT],OUTER_ARMOR[ZoneNames.LEFT_TORSO_REAR])
self.factory.set_arm_armor(OUTER_ARMOR[ZoneNames.LEFT_ARM])
self.factory.set_leg_armor(OUTER_ARMOR[ZoneNames.LEFT_LEG])