diff --git a/src/anndata/experimental/backed/_io.py b/src/anndata/experimental/backed/_io.py index 009ca982f..ef7782e95 100644 --- a/src/anndata/experimental/backed/_io.py +++ b/src/anndata/experimental/backed/_io.py @@ -9,7 +9,7 @@ from anndata._io.specs.registry import read_elem_lazy from anndata._types import AnnDataElem -from testing.anndata._pytest import doctest_needs +from testing.anndata._doctest import doctest_needs from ..._core.anndata import AnnData from ..._settings import settings diff --git a/src/testing/anndata/_doctest.py b/src/testing/anndata/_doctest.py new file mode 100644 index 000000000..cf4854ca0 --- /dev/null +++ b/src/testing/anndata/_doctest.py @@ -0,0 +1,16 @@ +from __future__ import annotations + +from collections.abc import Callable +from typing import TypeVar + +F = TypeVar("F", bound=Callable) + + +def doctest_needs(mod: str) -> Callable[[F], F]: + """Mark function with doctest dependency.""" + + def decorator(func: F) -> F: + func._doctest_needs = mod + return func + + return decorator diff --git a/src/testing/anndata/_pytest.py b/src/testing/anndata/_pytest.py index b557b1a8a..19ef860bf 100644 --- a/src/testing/anndata/_pytest.py +++ b/src/testing/anndata/_pytest.py @@ -16,11 +16,8 @@ import pytest if TYPE_CHECKING: - from collections.abc import Callable, Generator, Iterable + from collections.abc import Generator, Iterable from pathlib import Path - from typing import TypeVar - - F = TypeVar("F", bound=Callable) # TODO: Should be done in pyproject.toml eventually @@ -44,16 +41,6 @@ def _anndata_test_env(request: pytest.FixtureRequest) -> None: anndata.settings.reset(anndata.settings._registered_options.keys()) -def doctest_needs(mod: str) -> Callable[[F], F]: - """Mark function with doctest dependency.""" - - def decorator(func: F) -> F: - func._doctest_needs = mod - return func - - return decorator - - @pytest.fixture def _doctest_env( request: pytest.FixtureRequest, cache: pytest.Cache, tmp_path: Path