ENH: Add annotations for `<module>.test` objects

11 files changed, 31 insertions, 0 deletions

diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi
index ca0c79b17..e796dc15a 100644
--- a/numpy/__init__.pyi
+++ b/numpy/__init__.pyi
@@ -8,6 +8,7 @@ from abc import abstractmethod
 from types import TracebackType, MappingProxyType
 from contextlib import ContextDecorator
 
+from numpy._pytesttester import PytestTester
 from numpy.core.multiarray import flagsobj
 from numpy.core._internal import _ctypes
 from numpy.typing import (
@@ -610,6 +611,7 @@ __all__: List[str]
 __path__: List[str]
 __version__: str
 __git_version__: str
+test: PytestTester
 
 # TODO: Move placeholders to their respective module once
 # their annotations are properly implemented
diff --git a/numpy/f2py/__init__.pyi b/numpy/f2py/__init__.pyi
index b612ce15d..7d8e092ea 100644
--- a/numpy/f2py/__init__.pyi
+++ b/numpy/f2py/__init__.pyi
@@ -3,6 +3,8 @@ import subprocess
 from typing import Any, List, Iterable, Dict, overload
 from typing_extensions import TypedDict, Literal as L
 
+from numpy._pytesttester import PytestTester
+
 class _F2PyDictBase(TypedDict):
     csrc: List[str]
     h: List[str]
@@ -13,6 +15,7 @@ class _F2PyDict(_F2PyDictBase, total=False):
 
 __all__: List[str]
 __path__: List[str]
+test: PytestTester
 
 def run_main(comline_list: Iterable[str]) -> Dict[str, _F2PyDict]: ...
 
diff --git a/numpy/fft/__init__.pyi b/numpy/fft/__init__.pyi
index 3c191a35f..efdaf864d 100644
--- a/numpy/fft/__init__.pyi
+++ b/numpy/fft/__init__.pyi
@@ -1,6 +1,9 @@
 from typing import Any, List
 
+from numpy._pytesttester import PytestTester
+
 __all__: List[str]
+test: PytestTester
 
 def fft(a, n=..., axis=..., norm=...): ...
 def ifft(a, n=..., axis=..., norm=...): ...
diff --git a/numpy/lib/__init__.pyi b/numpy/lib/__init__.pyi
index 45a283782..50bcdd5a6 100644
--- a/numpy/lib/__init__.pyi
+++ b/numpy/lib/__init__.pyi
@@ -1,6 +1,8 @@
 import math as math
 from typing import Any, List
 
+from numpy._pytesttester import PytestTester
+
 from numpy import (
     ndenumerate as ndenumerate,
     ndindex as ndindex,
@@ -231,6 +233,7 @@ from numpy.core.multiarray import (
 )
 
 __all__: List[str]
+test: PytestTester
 
 __version__ = version
 emath = scimath
diff --git a/numpy/linalg/__init__.pyi b/numpy/linalg/__init__.pyi
index 00db05079..1a9028b9a 100644
--- a/numpy/linalg/__init__.pyi
+++ b/numpy/linalg/__init__.pyi
@@ -1,6 +1,9 @@
 from typing import Any, List
 
+from numpy._pytesttester import PytestTester
+
 __all__: List[str]
+test: PytestTester
 
 class LinAlgError(Exception): ...
 
diff --git a/numpy/ma/__init__.pyi b/numpy/ma/__init__.pyi
index a9a833e52..6b916fc45 100644
--- a/numpy/ma/__init__.pyi
+++ b/numpy/ma/__init__.pyi
@@ -1,5 +1,7 @@
 from typing import Any, List
 
+from numpy._pytesttester import PytestTester
+
 from numpy.ma import extras as extras
 
 from numpy.ma.core import (
@@ -230,3 +232,4 @@ from numpy.ma.extras import (
 )
 
 __all__: List[str]
+test: PytestTester
diff --git a/numpy/matrixlib/__init__.pyi b/numpy/matrixlib/__init__.pyi
index e4b5c19a2..5cb4d9c65 100644
--- a/numpy/matrixlib/__init__.pyi
+++ b/numpy/matrixlib/__init__.pyi
@@ -1,10 +1,13 @@
 from typing import Any, List
 
+from numpy._pytesttester import PytestTester
+
 from numpy import (
     matrix as matrix,
 )
 
 __all__: List[str]
+test: PytestTester
 
 def bmat(obj, ldict=..., gdict=...): ...
 def asmatrix(data, dtype=...): ...
diff --git a/numpy/polynomial/__init__.pyi b/numpy/polynomial/__init__.pyi
index bebedb3a6..ef8a93d95 100644
--- a/numpy/polynomial/__init__.pyi
+++ b/numpy/polynomial/__init__.pyi
@@ -1,5 +1,7 @@
 from typing import List
 
+from numpy._pytesttester import PytestTester
+
 from numpy.polynomial import (
     chebyshev as chebyshev,
     hermite as hermite,
@@ -16,5 +18,6 @@ from numpy.polynomial.legendre import Legendre as Legendre
 from numpy.polynomial.polynomial import Polynomial as Polynomial
 
 __all__: List[str]
+test: PytestTester
 
 def set_default_printstyle(style): ...
diff --git a/numpy/random/__init__.pyi b/numpy/random/__init__.pyi
index 48b620c48..5705bc4a7 100644
--- a/numpy/random/__init__.pyi
+++ b/numpy/random/__init__.pyi
@@ -1,5 +1,7 @@
 from typing import List
 
+from numpy._pytesttester import PytestTester
+
 from numpy.random._generator import Generator as Generator
 from numpy.random._generator import default_rng as default_rng
 from numpy.random._mt19937 import MT19937 as MT19937
@@ -66,3 +68,4 @@ from numpy.random.mtrand import (
 )
 
 __all__: List[str]
+test: PytestTester
diff --git a/numpy/testing/__init__.pyi b/numpy/testing/__init__.pyi
index 9eaba3deb..75854c0e3 100644
--- a/numpy/testing/__init__.pyi
+++ b/numpy/testing/__init__.pyi
@@ -1,5 +1,7 @@
 from typing import List
 
+from numpy._pytesttester import PytestTester
+
 from unittest import (
     TestCase as TestCase,
 )
@@ -47,6 +49,7 @@ from numpy.testing._private.utils import (
 )
 
 __all__: List[str]
+test: PytestTester
 
 def run_module_suite(
     file_to_run: None | str = ...,
diff --git a/numpy/typing/tests/data/reveal/modules.py b/numpy/typing/tests/data/reveal/modules.py
index b045585b2..7e695433e 100644
--- a/numpy/typing/tests/data/reveal/modules.py
+++ b/numpy/typing/tests/data/reveal/modules.py
@@ -32,6 +32,8 @@ reveal_type(np.polynomial.polynomial)  # E: ModuleType
 reveal_type(np.__path__)  # E: list[builtins.str]
 reveal_type(np.__version__)  # E: str
 reveal_type(np.__git_version__)  # E: str
+reveal_type(np.test)  # E: numpy._pytesttester.PytestTester
+reveal_type(np.test.module_name)  # E: str
 
 reveal_type(np.__all__)  # E: list[builtins.str]
 reveal_type(np.char.__all__)  # E: list[builtins.str]