Files
Grinch-AP/worlds/stardew_valley/test/assertion/goal_assert.py
Jérémie Bolduc 4ebabc1208 Stardew Valley: Move filler pool generation out of the world class (#4372)
* merge group options so specific handling is not needed when generating filler pool

* fix

* remove unneeded imports

* self review

* remove unneeded imports

* looks like typing was missing woopsi
2025-03-08 12:13:33 -05:00

56 lines
1.7 KiB
Python

from unittest import TestCase
from BaseClasses import MultiWorld
from .option_assert import get_stardew_options
from ... import options
def is_goal(multiworld: MultiWorld, goal: int) -> bool:
return get_stardew_options(multiworld).goal.value == goal
def is_bottom_mines(multiworld: MultiWorld) -> bool:
return is_goal(multiworld, options.Goal.option_bottom_of_the_mines)
def is_not_bottom_mines(multiworld: MultiWorld) -> bool:
return not is_bottom_mines(multiworld)
def is_walnut_hunter(multiworld: MultiWorld) -> bool:
return is_goal(multiworld, options.Goal.option_greatest_walnut_hunter)
def is_not_walnut_hunter(multiworld: MultiWorld) -> bool:
return not is_walnut_hunter(multiworld)
def is_perfection(multiworld: MultiWorld) -> bool:
return is_goal(multiworld, options.Goal.option_perfection)
def is_not_perfection(multiworld: MultiWorld) -> bool:
return not is_perfection(multiworld)
class GoalAssertMixin(TestCase):
def assert_ginger_island_is_included(self, multiworld: MultiWorld):
self.assertEqual(get_stardew_options(multiworld).exclude_ginger_island, options.ExcludeGingerIsland.option_false)
def assert_walnut_hunter_world_is_valid(self, multiworld: MultiWorld):
if is_not_walnut_hunter(multiworld):
return
self.assert_ginger_island_is_included(multiworld)
def assert_perfection_world_is_valid(self, multiworld: MultiWorld):
if is_not_perfection(multiworld):
return
self.assert_ginger_island_is_included(multiworld)
def assert_goal_world_is_valid(self, multiworld: MultiWorld):
self.assert_walnut_hunter_world_is_valid(multiworld)
self.assert_perfection_world_is_valid(multiworld)