Make SubTests importable from pytest_subtests (#122)

Fixes #121

---------

Co-authored-by: Bruno Oliveira <[email protected]>

CHANGELOG.rst

@@ -6,6 +6,7 @@ UNRELEASED
 
 * Python 3.12 is now officially supported (`#113`_).
 * Added typing support (`#115`_).
+* ``SubTests`` can be imported from ``pytest_subtests`` to type-annotate the ``subtests`` fixture.
 
 .. _#113: https://github.com/pytest-dev/pytest-subtests/pull/113
 .. _#115: https://github.com/pytest-dev/pytest-subtests/pull/115

src/pytest_subtests/__init__.py

@@ -0,0 +1,3 @@
+from .plugin import SubTests
+
+__all__ = ["SubTests"]

tests/test_subtests.py

@@ -138,6 +138,28 @@ def test_foo(subtests):
         expected_lines += ["* 1 passed, 3 xfailed, 2 subtests passed in *"]
         result.stdout.fnmatch_lines(expected_lines)
 
+    def test_typing_exported(
+        self, pytester: pytest.Pytester, mode: Literal["normal", "xdist"]
+    ) -> None:
+        pytester.makepyfile(
+            """
+            from pytest_subtests import SubTests
+
+            def test_typing_exported(subtests: SubTests) -> None:
+                assert isinstance(subtests, SubTests)
+            """
+        )
+        if mode == "normal":
+            result = pytester.runpytest()
+            expected_lines = ["collected 1 item"]
+        else:
+            assert mode == "xdist"
+            pytest.importorskip("xdist")
+            result = pytester.runpytest("-n1")
+            expected_lines = ["1 worker [1 item]"]
+        expected_lines += ["* 1 passed *"]
+        result.stdout.fnmatch_lines(expected_lines)
+
 
 class TestSubTest:
     """