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

allow set in var.contains (#3113)

Browse Source 
* allow set in var.contains

* fix UT
This commit is contained in:
Martin Xu 2024-04-18 20:00:18 -07:00 committed by GitHub
parent 98a4f58134
commit c567334c92
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
reflex/vars.py
View File

@ -1340,7 +1340,7 @@ class Var:
Returns:
A var representing the contain check.
"""
if not (types._issubclass(self._var_type, Union[dict, list, tuple, str])):
if not (types._issubclass(self._var_type, Union[dict, list, tuple, str, set])):
raise TypeError(
f"Var {self._var_full_name} of type {self._var_type} does not support contains check."
)

2
tests/test_var.py
View File

@ -415,6 +415,7 @@ def test_basic_operations(TestObj):
"var, expected",
[
(v([1, 2, 3]), "[1, 2, 3]"),
(v(set([1, 2, 3])), "[1, 2, 3]"),
(v(["1", "2", "3"]), '["1", "2", "3"]'),
(BaseVar(_var_name="foo", _var_type=list)._var_set_state("state"), "state.foo"),
(BaseVar(_var_name="foo", _var_type=list), "foo"),
@ -918,7 +919,6 @@ def test_unsupported_types_for_reverse(var):
BaseVar(_var_name="var", _var_type=int),
BaseVar(_var_name="var", _var_type=float),
BaseVar(_var_name="var", _var_type=bool),
BaseVar(_var_name="var", _var_type=set),
BaseVar(_var_name="var", _var_type=None),
],
)