109 lines
5.3 KiB
Python
109 lines
5.3 KiB
Python
from BaseClasses import CollectionState
|
|
|
|
from worlds.witness.test import WitnessMultiworldTestBase, WitnessTestBase
|
|
|
|
|
|
class TestMaxPanelHuntMinChecks(WitnessTestBase):
|
|
options = {
|
|
"victory_condition": "panel_hunt",
|
|
"panel_hunt_total": 100,
|
|
"panel_hunt_required_percentage": 100,
|
|
"panel_hunt_postgame": "disable_anything_locked_by_lasers",
|
|
"disable_non_randomized_puzzles": True,
|
|
"shuffle_discarded_panels": False,
|
|
"shuffle_vault_boxes": False,
|
|
}
|
|
|
|
def test_correct_panels_were_picked(self) -> None:
|
|
with self.subTest("Check that 100 Hunt Panels were actually picked."):
|
|
self.assertEqual(len(self.multiworld.find_item_locations("+1 Panel Hunt", self.player)), 100)
|
|
|
|
with self.subTest("Check that 100 Hunt Panels are enough"):
|
|
state_100 = CollectionState(self.multiworld)
|
|
panel_hunt_item = self.get_item_by_name("+1 Panel Hunt")
|
|
|
|
for _ in range(100):
|
|
state_100.collect(panel_hunt_item, True)
|
|
state_100.sweep_for_advancements([self.world.get_location("Tutorial Gate Open Solved")])
|
|
|
|
self.assertTrue(self.multiworld.completion_condition[self.player](state_100))
|
|
|
|
with self.subTest("Check that 99 Hunt Panels are not enough"):
|
|
state_99 = CollectionState(self.multiworld)
|
|
panel_hunt_item = self.get_item_by_name("+1 Panel Hunt")
|
|
|
|
for _ in range(99):
|
|
state_99.collect(panel_hunt_item, True)
|
|
state_99.sweep_for_advancements([self.world.get_location("Tutorial Gate Open Solved")])
|
|
|
|
self.assertFalse(self.multiworld.completion_condition[self.player](state_99))
|
|
|
|
|
|
class TestPanelHuntPostgame(WitnessMultiworldTestBase):
|
|
options_per_world = [
|
|
{
|
|
"panel_hunt_postgame": "everything_is_eligible"
|
|
},
|
|
{
|
|
"panel_hunt_postgame": "disable_mountain_lasers_locations"
|
|
},
|
|
{
|
|
"panel_hunt_postgame": "disable_challenge_lasers_locations"
|
|
},
|
|
{
|
|
"panel_hunt_postgame": "disable_anything_locked_by_lasers"
|
|
},
|
|
]
|
|
|
|
common_options = {
|
|
"victory_condition": "panel_hunt",
|
|
"panel_hunt_total": 40,
|
|
|
|
# Make sure we can check for Short vs Long Lasers locations by making Mountain Bottom Floor Discard accessible.
|
|
"shuffle_doors": "doors",
|
|
"shuffle_discarded_panels": True,
|
|
}
|
|
|
|
def test_panel_hunt_postgame(self) -> None:
|
|
for player_minus_one, options in enumerate(self.options_per_world):
|
|
player = player_minus_one + 1
|
|
postgame_option = options["panel_hunt_postgame"]
|
|
with self.subTest(f'Test that "{postgame_option}" results in 40 Hunt Panels.'):
|
|
self.assertEqual(len(self.multiworld.find_item_locations("+1 Panel Hunt", player)), 40)
|
|
|
|
# Test that the box gets extra checks from panel_hunt_postgame
|
|
|
|
with self.subTest('Test that "everything_is_eligible" has no Mountaintop Box Hunt Panels.'):
|
|
self.assert_location_does_not_exist("Mountaintop Box Short (Panel Hunt)", 1, strict_check=False)
|
|
self.assert_location_does_not_exist("Mountaintop Box Long (Panel Hunt)", 1, strict_check=False)
|
|
|
|
with self.subTest('Test that "disable_mountain_lasers_locations" has a Hunt Panel for Short, but not Long.'):
|
|
self.assert_location_exists("Mountaintop Box Short (Panel Hunt)", 2, strict_check=False)
|
|
self.assert_location_does_not_exist("Mountaintop Box Long (Panel Hunt)", 2, strict_check=False)
|
|
|
|
with self.subTest('Test that "disable_challenge_lasers_locations" has a Hunt Panel for Long, but not Short.'):
|
|
self.assert_location_does_not_exist("Mountaintop Box Short (Panel Hunt)", 3, strict_check=False)
|
|
self.assert_location_exists("Mountaintop Box Long (Panel Hunt)", 3, strict_check=False)
|
|
|
|
with self.subTest('Test that "disable_anything_locked_by_lasers" has both Mountaintop Box Hunt Panels.'):
|
|
self.assert_location_exists("Mountaintop Box Short (Panel Hunt)", 4, strict_check=False)
|
|
self.assert_location_exists("Mountaintop Box Long (Panel Hunt)", 4, strict_check=False)
|
|
|
|
# Check panel_hunt_postgame locations get disabled
|
|
|
|
with self.subTest('Test that "everything_is_eligible" does not disable any locked-by-lasers panels.'):
|
|
self.assert_location_exists("Mountain Floor 1 Right Row 5", 1)
|
|
self.assert_location_exists("Mountain Bottom Floor Discard", 1)
|
|
|
|
with self.subTest('Test that "disable_mountain_lasers_locations" disables only Shortbox-Locked panels.'):
|
|
self.assert_location_does_not_exist("Mountain Floor 1 Right Row 5", 2)
|
|
self.assert_location_exists("Mountain Bottom Floor Discard", 2)
|
|
|
|
with self.subTest('Test that "disable_challenge_lasers_locations" disables only Longbox-Locked panels.'):
|
|
self.assert_location_exists("Mountain Floor 1 Right Row 5", 3)
|
|
self.assert_location_does_not_exist("Mountain Bottom Floor Discard", 3)
|
|
|
|
with self.subTest('Test that "everything_is_eligible" disables only Shortbox-Locked panels.'):
|
|
self.assert_location_does_not_exist("Mountain Floor 1 Right Row 5", 4)
|
|
self.assert_location_does_not_exist("Mountain Bottom Floor Discard", 4)
|