33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
|
"""
|
||
|
|
Unit tests for item_groups.py
|
||
|
|
"""
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from ..item import item_groups, item_tables
|
||
|
|
|
||
|
|
|
||
|
|
class ItemGroupsUnitTests(unittest.TestCase):
|
||
|
|
def test_all_production_structure_groups_capture_all_units(self) -> None:
|
||
|
|
self.assertCountEqual(
|
||
|
|
item_groups.terran_units,
|
||
|
|
item_groups.barracks_units + item_groups.factory_units + item_groups.starport_units + item_groups.terran_mercenaries
|
||
|
|
)
|
||
|
|
self.assertCountEqual(
|
||
|
|
item_groups.protoss_units,
|
||
|
|
item_groups.gateway_units + item_groups.robo_units + item_groups.stargate_units
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_terran_original_progressive_group_fully_contained_in_wol_upgrades(self) -> None:
|
||
|
|
for item_name in item_groups.terran_original_progressive_upgrades:
|
||
|
|
self.assertIn(item_tables.item_table[item_name].type, (
|
||
|
|
item_tables.TerranItemType.Progressive, item_tables.TerranItemType.Progressive_2), f"{item_name} is not progressive")
|
||
|
|
self.assertIn(item_name, item_groups.wol_upgrades)
|
||
|
|
|
||
|
|
def test_all_items_in_stimpack_group_are_stimpacks(self) -> None:
|
||
|
|
for item_name in item_groups.terran_stimpacks:
|
||
|
|
self.assertIn("Stimpack", item_name)
|
||
|
|
|
||
|
|
def test_all_item_group_names_have_a_group_defined(self) -> None:
|
||
|
|
for display_name in item_groups.ItemGroupNames.get_all_group_names():
|
||
|
|
self.assertIn(display_name, item_groups.item_name_groups)
|