escape imported types in quotes

Author: miguelgrinberg

elasticsearch_dsl/wrappers.py

@@ -38,7 +38,7 @@
 from .utils import AttrDict
 
 ComparisonOperators: TypeAlias = Literal["lt", "lte", "gt", "gte"]
-RangeValT = TypeVar("RangeValT", bound=_SupportsComparison)
+RangeValT = TypeVar("RangeValT", bound="_SupportsComparison")
 
 __all__ = ["Range"]
 
@@ -47,7 +47,7 @@ class Range(AttrDict[ComparisonOperators, RangeValT]):
     OPS: ClassVar[
         Mapping[
             ComparisonOperators,
-            Callable[[_SupportsComparison, _SupportsComparison], bool],
+            Callable[["_SupportsComparison", "_SupportsComparison"], bool],
         ]
     ] = {
         "lt": operator.lt,
@@ -97,7 +97,7 @@ def __contains__(self, item: object) -> bool:
 
         # Cast to tell mypy whe have checked it and its ok to use the comparison methods
         # on `item`
-        item = cast(_SupportsComparison, item)
+        item = cast("_SupportsComparison", item)
 
         for op in self.OPS:
             if op in self._d_ and not self.OPS[op](item, self._d_[op]):

tests/test_wrappers.py

@@ -39,7 +39,7 @@
     ],
 )
 def test_range_contains(
-    kwargs: Mapping[str, _SupportsComparison], item: _SupportsComparison
+    kwargs: Mapping[str, "_SupportsComparison"], item: "_SupportsComparison"
 ) -> None:
     assert item in Range(**kwargs)
 
@@ -55,7 +55,7 @@ def test_range_contains(
     ],
 )
 def test_range_not_contains(
-    kwargs: Mapping[str, _SupportsComparison], item: _SupportsComparison
+    kwargs: Mapping[str, "_SupportsComparison"], item: "_SupportsComparison"
 ) -> None:
     assert item not in Range(**kwargs)
 
@@ -71,7 +71,7 @@ def test_range_not_contains(
     ],
 )
 def test_range_raises_value_error_on_wrong_params(
-    args: Sequence[Any], kwargs: Mapping[str, _SupportsComparison]
+    args: Sequence[Any], kwargs: Mapping[str, "_SupportsComparison"]
 ) -> None:
     with pytest.raises(ValueError):
         Range(*args, **kwargs)
@@ -87,8 +87,8 @@ def test_range_raises_value_error_on_wrong_params(
     ],
 )
 def test_range_lower(
-    range: Range[_SupportsComparison],
-    lower: Optional[_SupportsComparison],
+    range: Range["_SupportsComparison"],
+    lower: Optional["_SupportsComparison"],
     inclusive: bool,
 ) -> None:
     assert (lower, inclusive) == range.lower
@@ -104,8 +104,8 @@ def test_range_lower(
     ],
 )
 def test_range_upper(
-    range: Range[_SupportsComparison],
-    upper: Optional[_SupportsComparison],
+    range: Range["_SupportsComparison"],
+    upper: Optional["_SupportsComparison"],
     inclusive: bool,
 ) -> None:
     assert (upper, inclusive) == range.upper