88 lines
5.2 KiB
Python
88 lines
5.2 KiB
Python
from typing import Dict
|
|
|
|
from BaseClasses import MultiWorld
|
|
from Options import NamedRange
|
|
from .option_names import options_to_include
|
|
from .checks.world_checks import assert_can_win, assert_same_number_items_locations
|
|
from . import DLCQuestTestBase, setup_dlc_quest_solo_multiworld
|
|
from ... import AutoWorldRegister
|
|
|
|
|
|
def basic_checks(tester: DLCQuestTestBase, multiworld: MultiWorld):
|
|
assert_can_win(tester, multiworld)
|
|
assert_same_number_items_locations(tester, multiworld)
|
|
|
|
|
|
def get_option_choices(option) -> Dict[str, int]:
|
|
if issubclass(option, NamedRange):
|
|
return option.special_range_names
|
|
elif option.options:
|
|
return option.options
|
|
return {}
|
|
|
|
|
|
class TestGenerateDynamicOptions(DLCQuestTestBase):
|
|
def test_given_option_pair_when_generate_then_basic_checks(self):
|
|
num_options = len(options_to_include)
|
|
for option1_index in range(0, num_options):
|
|
for option2_index in range(option1_index + 1, num_options):
|
|
option1 = options_to_include[option1_index]
|
|
option2 = options_to_include[option2_index]
|
|
option1_choices = get_option_choices(option1)
|
|
option2_choices = get_option_choices(option2)
|
|
for key1 in option1_choices:
|
|
for key2 in option2_choices:
|
|
with self.subTest(f"{option1.internal_name}: {key1}, {option2.internal_name}: {key2}"):
|
|
choices = {option1.internal_name: option1_choices[key1],
|
|
option2.internal_name: option2_choices[key2]}
|
|
multiworld = setup_dlc_quest_solo_multiworld(choices)
|
|
basic_checks(self, multiworld)
|
|
|
|
def test_given_option_truple_when_generate_then_basic_checks(self):
|
|
num_options = len(options_to_include)
|
|
for option1_index in range(0, num_options):
|
|
for option2_index in range(option1_index + 1, num_options):
|
|
for option3_index in range(option2_index + 1, num_options):
|
|
option1 = options_to_include[option1_index]
|
|
option2 = options_to_include[option2_index]
|
|
option3 = options_to_include[option3_index]
|
|
option1_choices = get_option_choices(option1)
|
|
option2_choices = get_option_choices(option2)
|
|
option3_choices = get_option_choices(option3)
|
|
for key1 in option1_choices:
|
|
for key2 in option2_choices:
|
|
for key3 in option3_choices:
|
|
with self.subTest(f"{option1.internal_name}: {key1}, {option2.internal_name}: {key2}, {option3.internal_name}: {key3}"):
|
|
choices = {option1.internal_name: option1_choices[key1],
|
|
option2.internal_name: option2_choices[key2],
|
|
option3.internal_name: option3_choices[key3]}
|
|
multiworld = setup_dlc_quest_solo_multiworld(choices)
|
|
basic_checks(self, multiworld)
|
|
|
|
def test_given_option_quartet_when_generate_then_basic_checks(self):
|
|
num_options = len(options_to_include)
|
|
for option1_index in range(0, num_options):
|
|
for option2_index in range(option1_index + 1, num_options):
|
|
for option3_index in range(option2_index + 1, num_options):
|
|
for option4_index in range(option3_index + 1, num_options):
|
|
option1 = options_to_include[option1_index]
|
|
option2 = options_to_include[option2_index]
|
|
option3 = options_to_include[option3_index]
|
|
option4 = options_to_include[option4_index]
|
|
option1_choices = get_option_choices(option1)
|
|
option2_choices = get_option_choices(option2)
|
|
option3_choices = get_option_choices(option3)
|
|
option4_choices = get_option_choices(option4)
|
|
for key1 in option1_choices:
|
|
for key2 in option2_choices:
|
|
for key3 in option3_choices:
|
|
for key4 in option4_choices:
|
|
with self.subTest(
|
|
f"{option1.internal_name}: {key1}, {option2.internal_name}: {key2}, {option3.internal_name}: {key3}, {option4.internal_name}: {key4}"):
|
|
choices = {option1.internal_name: option1_choices[key1],
|
|
option2.internal_name: option2_choices[key2],
|
|
option3.internal_name: option3_choices[key3],
|
|
option4.internal_name: option4_choices[key4]}
|
|
multiworld = setup_dlc_quest_solo_multiworld(choices)
|
|
basic_checks(self, multiworld)
|