Some unit-test related fixes

Ziktofel · Jan 7, 2024 · 91ff4f1 · 91ff4f1
1 parent 37adf99
commit 91ff4f1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/worlds/sc2/PoolFilter.py b/worlds/sc2/PoolFilter.py
@@ -436,7 +436,7 @@ def attempt_removal(item: Item) -> bool:
             inventory = [item for item in inventory if not item.name.endswith("(Dark Templar/Avenger/Blood Hunter)")]
         if not {ItemNames.HIGH_TEMPLAR, ItemNames.SIGNIFIER, ItemNames.ASCENDANT, ItemNames.DARK_TEMPLAR} & logical_inventory_set:
             inventory = [item for item in inventory if not item.name.endswith("(Archon)")]
-            logical_inventory_set.difference_update((item_name for item_name in logical_inventory_set if item_name.endswith("(Archon)")))
+            logical_inventory_set.difference_update([item_name for item_name in logical_inventory_set if item_name.endswith("(Archon)")])
         if not {ItemNames.HIGH_TEMPLAR, ItemNames.SIGNIFIER, ItemNames.ARCHON_HIGH_ARCHON} & logical_inventory_set:
             inventory = [item for item in inventory if not item.name.endswith("(High Templar/Signifier)")]
         if ItemNames.SUPPLICANT not in logical_inventory_set:

diff --git a/worlds/sc2/test/test_Regions.py b/worlds/sc2/test/test_Regions.py
@@ -31,7 +31,8 @@ class TestGridGeneration(Sc2TestBase):
         "enable_prophecy_missions": True,
         "enable_lotv_prologue_missions": False,
         "enable_lotv_missions": False,
-        "enable_epilogue_missions": False
+        "enable_epilogue_missions": False,
+        "enable_nco_missions": False
     }
 
     def test_size_matches_exclusions(self):