Skip to content

Commit

Permalink
sc2: Updated war council generation unit test to check start inventory
Browse files Browse the repository at this point in the history
  • Loading branch information
MatthewMarinets committed Jul 28, 2024
1 parent 0779a21 commit 98bebf9
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion worlds/sc2/test/test_generation.py
Original file line number Diff line number Diff line change
Expand Up @@ -472,7 +472,7 @@ def test_planetary_orbital_module_not_present_without_cc_spells(self) -> None:
self.assertIn(item_names.PLANETARY_FORTRESS, itempool)
self.assertNotIn(item_names.PLANETARY_FORTRESS_ORBITAL_MODULE, itempool)

def test_disabling_unit_nerfs_removes_war_council_upgrades(self) -> None:
def test_disabling_unit_nerfs_start_inventories_war_council_upgrades(self) -> None:
world_options = {
'enable_wol_missions': False,
'enable_prophecy_missions': True,
Expand All @@ -489,6 +489,9 @@ def test_disabling_unit_nerfs_removes_war_council_upgrades(self) -> None:
itempool = [item.name for item in self.multiworld.itempool]
war_council_item_names = set(item_groups.item_name_groups[item_groups.ItemGroupNames.WAR_COUNCIL])
present_war_council_items = war_council_item_names.intersection(itempool)
starting_inventory = [item.name for item in self.multiworld.precollected_items[self.player]]
starting_war_council_items = war_council_item_names.intersection(starting_inventory)

self.assertTrue(itempool)
self.assertFalse(present_war_council_items, f'Found war council upgrades when nerf_unit_baselines is false: {present_war_council_items}')
self.assertEqual(war_council_item_names, starting_war_council_items)

0 comments on commit 98bebf9

Please sign in to comment.