58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
import unittest
|
|
|
|
from mech_factory import MechFabrik
|
|
from mech_factory.components 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,
|
|
}
|
|
|
|
OUTER_ARMOR = {
|
|
ZoneNames.HEAD: 6,
|
|
ZoneNames.MIDDLE_TORSO_FRONT: 9,
|
|
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 AbstractTestMechAmor(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
fabrik = MechFabrik('Heuschrecke', 20.0, 8)
|
|
|
|
fabrik.set_head_armor(OUTER_ARMOR[ZoneNames.HEAD])
|
|
fabrik.set_middle_torso_armor(OUTER_ARMOR[ZoneNames.MIDDLE_TORSO_FRONT],OUTER_ARMOR[ZoneNames.MIDDLE_TORSO_REAR])
|
|
fabrik.set_side_torso_armor(OUTER_ARMOR[ZoneNames.LEFT_TORSO_FRONT],OUTER_ARMOR[ZoneNames.LEFT_TORSO_REAR])
|
|
fabrik.set_arm_armor(OUTER_ARMOR[ZoneNames.LEFT_ARM])
|
|
fabrik.set_leg_armor(OUTER_ARMOR[ZoneNames.LEFT_LEG])
|
|
|
|
self.mech = fabrik.auslieferung()
|