Skip to content

Commit e987789

Browse files
committed
refactor: align state proxies with stubs and remove typing.cast
1 parent ec63e1b commit e987789

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

src/_algopy_testing/state/global_state.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,11 +120,11 @@ def value(self) -> None:
120120
def __bool__(self) -> bool:
121121
return self._key is not None or self._pending_value is not None
122122

123-
def get(self, default: _T | None = None) -> _T:
123+
def get(self, default: _T) -> _T:
124124
try:
125125
return self.value
126126
except ValueError:
127-
return typing.cast(_T, default)
127+
return default
128128

129129
def maybe(self) -> tuple[_T | None, bool]:
130130
try:

src/_algopy_testing/state/local_state.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,12 +70,12 @@ def __contains__(self, key: algopy.Account | algopy.UInt64 | int) -> bool:
7070
return False
7171
return True
7272

73-
def get(self, key: algopy.Account | algopy.UInt64 | int, default: _T | None = None) -> _T:
73+
def get(self, key: algopy.Account | algopy.UInt64 | int, default: _T) -> _T:
7474
account = _get_account(key)
7575
try:
7676
return self[account]
7777
except KeyError:
78-
return typing.cast(_T, default)
78+
return default
7979

8080
def maybe(self, key: algopy.Account | algopy.UInt64 | int) -> tuple[_T, bool]:
8181
account = _get_account(key)

tests/state/test_global_state.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ def test_get_method(
226226
assert gs.get(default=None) is None
227227

228228
gs.value = value
229-
assert gs.get() == value
229+
assert gs.get(default=None) == value
230230

231231
@pytest.mark.parametrize(("type_", "value", "expected_value"), *test_data_array)
232232
def test_maybe_method(

tests/state/test_local_state.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def test_get_method(self, context: AlgopyTestContext, type_: Any, value: Any) ->
120120
assert ls.get(context.default_sender, default=None) is None
121121

122122
ls[context.default_sender] = value
123-
assert ls.get(context.default_sender) == value
123+
assert ls.get(context.default_sender, default=None) == value
124124

125125
@pytest.mark.parametrize(("type_", "value"), *test_data_array)
126126
def test_maybe_method(self, context: AlgopyTestContext, type_: Any, value: Any) -> None:

0 commit comments

Comments
 (0)