Make test functions private

Author: crusaderky

docs/api-reference.md

@@ -15,15 +15,3 @@
     setdiff1d
     sinc
 ```
-
-## Test tools
-
-```{eval-rst}
-.. currentmodule:: array_api_extra.testing
-.. autosummary::
-    :nosignatures:
-    :toctree: generated
-
-    xp_assert_equal
-    xp_assert_close
-```

src/array_api_extra/testing.py renamed to src/array_api_extra/_lib/_testing.py

@@ -1,12 +1,16 @@
-"""Testing utilities."""
+"""
+Testing utilities.
 
-from ._lib._compat import (
+Note that this is private API; don't expect it to be stable.
+"""
+
+from ._compat import (
     array_namespace,
     is_cupy_namespace,
     is_pydata_sparse_namespace,
     is_torch_namespace,
 )
-from ._lib._typing import Array, ModuleType
+from ._typing import Array, ModuleType
 
 __all__ = ["xp_assert_close", "xp_assert_equal"]
 

tests/test_at.py

@@ -12,8 +12,8 @@
 
 from array_api_extra import at
 from array_api_extra._funcs import _AtOp
+from array_api_extra._lib._testing import xp_assert_equal
 from array_api_extra._lib._typing import Array, ModuleType
-from array_api_extra.testing import xp_assert_equal
 
 from .conftest import Library
 

tests/test_funcs.py

@@ -16,8 +16,8 @@
     sinc,
 )
 from array_api_extra._lib._compat import device as get_device
+from array_api_extra._lib._testing import xp_assert_close, xp_assert_equal
 from array_api_extra._lib._typing import Array, Device, ModuleType
-from array_api_extra.testing import xp_assert_close, xp_assert_equal
 
 from .conftest import Library
 

tests/test_testing.py

@@ -1,7 +1,7 @@
 import numpy as np
 import pytest
 
-from array_api_extra.testing import xp_assert_close, xp_assert_equal
+from array_api_extra._lib._testing import xp_assert_close, xp_assert_equal
 
 from .conftest import Library
 

tests/test_utils.py

@@ -2,9 +2,9 @@
 import pytest
 
 from array_api_extra._lib._compat import device as get_device
+from array_api_extra._lib._testing import xp_assert_equal
 from array_api_extra._lib._typing import Array, Device, ModuleType
 from array_api_extra._lib._utils import in1d
-from array_api_extra.testing import xp_assert_equal
 
 from .conftest import Library