laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Benedikt Bartscher 2024-11-28 21:57:50 +01:00
parent 5be7ce8a0d
commit d93fae6a7c
No known key found for this signature in database
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/units/test_state.py
View File

@ -55,7 +55,11 @@ from reflex.state import (
) )
from reflex.testing import chdir from reflex.testing import chdir
from reflex.utils import format, prerequisites, types from reflex.utils import format, prerequisites, types
from reflex.utils.exceptions import ReflexRuntimeError, SetUndefinedStateVarError from reflex.utils.exceptions import (
ReflexRuntimeError,
SetUndefinedStateVarError,
StateSerializationError,
)
from reflex.utils.format import json_dumps from reflex.utils.format import json_dumps
from reflex.vars.base import Var, computed_var from reflex.vars.base import Var, computed_var
from tests.units.states.mutation import MutableSQLAModel, MutableTestState from tests.units.states.mutation import MutableSQLAModel, MutableTestState
@ -3433,8 +3437,9 @@ def test_fallback_pickle():
# Some object, like generator, are still unpicklable with dill. # Some object, like generator, are still unpicklable with dill.
state3 = DillState(_reflex_internal_init=True) # type: ignore state3 = DillState(_reflex_internal_init=True) # type: ignore
state3._g = (i for i in range(10)) state3._g = (i for i in range(10))
pk3 = state3._serialize()
assert len(pk3) == 0 with pytest.raises(StateSerializationError):
_ = state3._serialize()
def test_typed_state() -> None: def test_typed_state() -> None: