fix(pyright): 'Run pyright on 'tests' using the installed stubs'

Author: cmp0xff

pandas-stubs/_config/config.pyi

@@ -182,3 +182,5 @@ class option_context(ContextDecorator):
     def __exit__(self, *args: object) -> None: ...
 
 class OptionError(AttributeError, KeyError): ...
+
+__all__ = ["DictWrapper", "Display", "Options"]

pandas-stubs/_testing/__init__.pyi

@@ -187,3 +187,5 @@ def assert_produces_warning(
 def ensure_clean(
     filename: str | None = ..., return_filelike: bool = ..., **kwargs: Any
 ) -> Generator[str, None, None]: ...
+
+__all__ = ["ensure_clean"]

pandas-stubs/_typing.pyi

@@ -842,4 +842,29 @@ IntoColumn: TypeAlias = (
     AnyArrayLike | Scalar | Callable[[DataFrame], AnyArrayLike | Scalar]
 )
 
-__all__ = ["npt", "type_t"]
+__all__ = [
+    "npt",
+    "type_t",
+    "T",
+    "TakeIndexer",
+    "type_t",
+    "DtypeObj",
+    "Scalar",
+    "IntervalClosedType",
+    "TimeUnit",
+    "np_ndarray_bool",
+    "BooleanDtypeArg",
+    "BytesDtypeArg",
+    "CategoryDtypeArg",
+    "FloatDtypeArg",
+    "IntDtypeArg",
+    "ObjectDtypeArg",
+    "StrDtypeArg",
+    "TimedeltaDtypeArg",
+    "TimestampDtypeArg",
+    "UIntDtypeArg",
+    "VoidDtypeArg",
+    "np_ndarray_int",
+    "FulldatetimeDict",
+    "TimeUnit",
+]

pandas-stubs/io/json/_json.pyi

@@ -238,3 +238,5 @@ class JsonReader(abc.Iterator, Generic[NDFrameT]):
         exc_value: BaseException | None,
         traceback: TracebackType | None,
     ) -> None: ...
+
+__all__ = ["read_json", "JsonReader"]

tests/test_scalars.py

@@ -580,7 +580,7 @@ def test_timedelta_add_sub() -> None:
     )
     check(
         assert_type(  # type: ignore[assert-type]
-            as_timedelta64 + td,  # pyright: ignore[reportAssertTypeFailure]
+            as_timedelta64 + td,
             pd.Timedelta,
         ),
         pd.Timedelta,
@@ -645,7 +645,7 @@ def test_timedelta_add_sub() -> None:
     )
     check(
         assert_type(  # type: ignore[assert-type]
-            as_timedelta64 - td,  # pyright: ignore[reportAssertTypeFailure]
+            as_timedelta64 - td,
             pd.Timedelta,
         ),
         pd.Timedelta,