DLCQuest: Use options API for campaign and remove unused imports in tests #4181

This commit is contained in:
Nicholas Saylor 2024-11-12 18:13:49 -05:00 committed by GitHub
parent 17f03bb5f8
commit 8b87e20a96
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 5 deletions

View File

@ -5,7 +5,6 @@ from Options import NamedRange
from .option_names import options_to_include from .option_names import options_to_include
from .checks.world_checks import assert_can_win, assert_same_number_items_locations from .checks.world_checks import assert_can_win, assert_same_number_items_locations
from . import DLCQuestTestBase, setup_dlc_quest_solo_multiworld from . import DLCQuestTestBase, setup_dlc_quest_solo_multiworld
from ... import AutoWorldRegister
def basic_checks(tester: DLCQuestTestBase, multiworld: MultiWorld): def basic_checks(tester: DLCQuestTestBase, multiworld: MultiWorld):

View File

@ -1,6 +1,6 @@
from typing import List from typing import List
from BaseClasses import MultiWorld, ItemClassification from BaseClasses import MultiWorld
from .. import DLCQuestTestBase from .. import DLCQuestTestBase
from ... import Options from ... import Options
@ -14,7 +14,7 @@ def get_all_location_names(multiworld: MultiWorld) -> List[str]:
def assert_victory_exists(tester: DLCQuestTestBase, multiworld: MultiWorld): def assert_victory_exists(tester: DLCQuestTestBase, multiworld: MultiWorld):
campaign = multiworld.campaign[1] campaign = multiworld.worlds[1].options.campaign
all_items = [item.name for item in multiworld.get_items()] all_items = [item.name for item in multiworld.get_items()]
if campaign == Options.Campaign.option_basic or campaign == Options.Campaign.option_both: if campaign == Options.Campaign.option_basic or campaign == Options.Campaign.option_both:
tester.assertIn("Victory Basic", all_items) tester.assertIn("Victory Basic", all_items)
@ -25,7 +25,7 @@ def assert_victory_exists(tester: DLCQuestTestBase, multiworld: MultiWorld):
def collect_all_then_assert_can_win(tester: DLCQuestTestBase, multiworld: MultiWorld): def collect_all_then_assert_can_win(tester: DLCQuestTestBase, multiworld: MultiWorld):
for item in multiworld.get_items(): for item in multiworld.get_items():
multiworld.state.collect(item) multiworld.state.collect(item)
campaign = multiworld.campaign[1] campaign = multiworld.worlds[1].options.campaign
if campaign == Options.Campaign.option_basic or campaign == Options.Campaign.option_both: if campaign == Options.Campaign.option_basic or campaign == Options.Campaign.option_both:
tester.assertTrue(multiworld.find_item("Victory Basic", 1).can_reach(multiworld.state)) tester.assertTrue(multiworld.find_item("Victory Basic", 1).can_reach(multiworld.state))
if campaign == Options.Campaign.option_live_freemium_or_die or campaign == Options.Campaign.option_both: if campaign == Options.Campaign.option_live_freemium_or_die or campaign == Options.Campaign.option_both:
@ -39,4 +39,4 @@ def assert_can_win(tester: DLCQuestTestBase, multiworld: MultiWorld):
def assert_same_number_items_locations(tester: DLCQuestTestBase, multiworld: MultiWorld): def assert_same_number_items_locations(tester: DLCQuestTestBase, multiworld: MultiWorld):
non_event_locations = [location for location in multiworld.get_locations() if not location.advancement] non_event_locations = [location for location in multiworld.get_locations() if not location.advancement]
tester.assertEqual(len(multiworld.itempool), len(non_event_locations)) tester.assertEqual(len(multiworld.itempool), len(non_event_locations))