54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
# Add root folder to python paths
|
|
# This must be done on every test in order to pass in Travis
|
|
import os
|
|
import sys
|
|
|
|
script_dir = os.path.dirname(os.path.abspath(__file__))
|
|
sys.path.append(os.path.realpath(os.path.join(script_dir, '..', '..', '..', '..')))
|
|
|
|
import pytest
|
|
from eos.utils.stats import DmgTypes, RRTypes
|
|
|
|
|
|
@pytest.fixture()
|
|
def setup_damage_types():
|
|
return DmgTypes(10, 20, 30, 40)
|
|
|
|
|
|
def test_dmgtypes_names():
|
|
assert DmgTypes.names() == ['em', 'thermal', 'kinetic', 'explosive']
|
|
assert DmgTypes.names(True) == ['em', 'th', 'kin', 'exp']
|
|
assert DmgTypes.names(short=True) == ['em', 'th', 'kin', 'exp']
|
|
|
|
|
|
def test_dmgtypes__repr(setup_damage_types):
|
|
assert setup_damage_types.__repr__() == '<DmgTypes(em=10, thermal=20, kinetic=30, explosive=40, total=100)>'
|
|
|
|
|
|
def test_dmgtypes_names_lambda():
|
|
assert DmgTypes.names(False, lambda v: v.capitalize()) == ['Em', 'Thermal', 'Kinetic', 'Explosive']
|
|
assert DmgTypes.names(True, lambda v: v.upper()) == ['EM', 'TH', 'KIN', 'EXP']
|
|
|
|
|
|
@pytest.fixture()
|
|
def setup_rr_types():
|
|
return RRTypes(10, 20, 30, 40)
|
|
|
|
|
|
def test_rrtypes_names():
|
|
assert RRTypes.names() == ['shield', 'armor', 'hull']
|
|
assert RRTypes.names(True) == ['shield', 'armor', 'hull']
|
|
assert RRTypes.names(ehpOnly=True) == ['shield', 'armor', 'hull']
|
|
assert RRTypes.names(False) == ['shield', 'armor', 'hull', 'capacitor']
|
|
|
|
|
|
def test_rrtypes__repr(setup_rr_types):
|
|
assert setup_rr_types.__repr__() == '<RRTypes(shield=10, armor=20, hull=30, capacitor=40)>'
|
|
|
|
|
|
def test_rrtypes_names_lambda():
|
|
assert RRTypes.names(True, lambda v: v.capitalize()) == ['Shield', 'Armor', 'Hull']
|
|
assert RRTypes.names(postProcessor=lambda v: v.upper(), ehpOnly=False) == ['SHIELD', 'ARMOR', 'HULL', 'CAPACITOR']
|
|
|
|
|