From ecb22642af291e05bdc6ae729bb14d4d1ae83792 Mon Sep 17 00:00:00 2001 From: Duck <31627079+duckboycool@users.noreply.github.com> Date: Fri, 8 Aug 2025 16:24:19 -0600 Subject: [PATCH] Tests: Handle optional args for `get_all_state` patch (#5297) * Make `use_cache` optional * Pass all kwargs --- test/general/test_entrances.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/general/test_entrances.py b/test/general/test_entrances.py index 88362c8f..79025534 100644 --- a/test/general/test_entrances.py +++ b/test/general/test_entrances.py @@ -48,13 +48,14 @@ class TestBase(unittest.TestCase): original_get_all_state = multiworld.get_all_state - def patched_get_all_state(use_cache: bool, allow_partial_entrances: bool = False): + def patched_get_all_state(use_cache: bool | None = None, allow_partial_entrances: bool = False, + **kwargs): self.assertTrue(allow_partial_entrances, ( "Before the connect_entrances step finishes, other worlds might still have partial entrances. " "As such, any call to get_all_state must use allow_partial_entrances = True." )) - return original_get_all_state(use_cache, allow_partial_entrances) + return original_get_all_state(use_cache, allow_partial_entrances, **kwargs) multiworld.get_all_state = patched_get_all_state