diff options
Diffstat (limited to 'test/__init__.py')
| -rw-r--r-- | test/__init__.py | 8 |
1 files changed, 6 insertions, 2 deletions
| diff --git a/test/__init__.py b/test/__init__.py index ffbf903..7ff456d 100644 --- a/test/__init__.py +++ b/test/__init__.py | |||
| @@ -1,7 +1,6 @@ | |||
| 1 | from typing import ClassVar | 1 | from typing import ClassVar |
| 2 | 2 | ||
| 3 | from test.bases import WorldTestBase | 3 | from test.bases import WorldTestBase |
| 4 | from .. import LingoTestOptions | ||
| 5 | 4 | ||
| 6 | 5 | ||
| 7 | class LingoTestBase(WorldTestBase): | 6 | class LingoTestBase(WorldTestBase): |
| @@ -9,5 +8,10 @@ class LingoTestBase(WorldTestBase): | |||
| 9 | player: ClassVar[int] = 1 | 8 | player: ClassVar[int] = 1 |
| 10 | 9 | ||
| 11 | def world_setup(self, *args, **kwargs): | 10 | def world_setup(self, *args, **kwargs): |
| 12 | LingoTestOptions.disable_forced_good_item = True | ||
| 13 | super().world_setup(*args, **kwargs) | 11 | super().world_setup(*args, **kwargs) |
| 12 | |||
| 13 | def remove_forced_good_item(self): | ||
| 14 | location = self.multiworld.get_location("Second Room - Good Luck", self.player) | ||
| 15 | self.remove(location.item) | ||
| 16 | self.multiworld.itempool.append(location.item) | ||
| 17 | self.multiworld.state.events.add(location) | ||
