mirror of
https://github.com/GoodStartLabs/AI_Diplomacy.git
synced 2026-04-19 12:58:09 +00:00
540c2003e8
The PowerEnum correctly handles some misspellings. It can be easily expanded to handle more within the _POWER_ALIASES dict.
21 lines
731 B
Python
21 lines
731 B
Python
from models import PowerEnum
|
|
|
|
|
|
def test_power_name_aliases():
|
|
# Test all aliases defined in _POWER_ALIASES
|
|
assert PowerEnum("UK") == PowerEnum.ENGLAND
|
|
assert PowerEnum("BRIT") == PowerEnum.ENGLAND
|
|
assert PowerEnum("EGMANY") == PowerEnum.GERMANY
|
|
assert PowerEnum("GERMAN") == PowerEnum.GERMANY
|
|
|
|
# Test direct enum values (no alias needed)
|
|
assert PowerEnum("AUSTRIA") == PowerEnum.AUSTRIA
|
|
assert PowerEnum("FRANCE") == PowerEnum.FRANCE
|
|
|
|
# Test case insensitivity
|
|
assert PowerEnum("france") == PowerEnum.FRANCE
|
|
assert PowerEnum("iTaLy") == PowerEnum.ITALY
|
|
|
|
# Test with whitespace
|
|
assert PowerEnum(" RUSSIA ") == PowerEnum.RUSSIA
|
|
assert PowerEnum("TURKEY ") == PowerEnum.TURKEY
|