Skip to content

Commit 1b23faa

Browse files
committed
feat: add _assert_literal_predicate_type for In and NotIn
1 parent 1a3e702 commit 1b23faa

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/expressions/test_expressions.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1224,6 +1224,9 @@ def _assert_literal_predicate_type(expr: LiteralPredicate[L]) -> None:
12241224

12251225

12261226
_assert_literal_predicate_type(EqualTo("a", "b"))
1227+
_assert_literal_predicate_type(In("a", ("a", "b", "c")))
1228+
_assert_literal_predicate_type(In("a", (1, 2, 3)))
1229+
_assert_literal_predicate_type(NotIn("a", ("a", "b", "c")))
12271230
assert_type(In("a", ("a", "b", "c")), In[str])
12281231
assert_type(In("a", (1, 2, 3)), In[int])
12291232
assert_type(NotIn("a", ("a", "b", "c")), NotIn[str])

0 commit comments

Comments
 (0)