Overhaul test_all

crusaderky · crusaderky · commit d6a53f91583c · 2025-04-21T11:34:47.000+01:00
diff --git a/array_api_compat/common/_aliases.py b/array_api_compat/common/_aliases.py
@@ -720,8 +720,6 @@ def iinfo(type_: DType | Array, /, xp: Namespace) -> Any:
     "finfo",
     "iinfo",
 ]
-_all_ignore = ["inspect", "array_namespace", "NamedTuple"]
-
 
 def __dir__() -> list[str]:
     return __all__
diff --git a/array_api_compat/common/_helpers.py b/array_api_compat/common/_helpers.py
@@ -1042,7 +1042,5 @@ def is_lazy_array(x: object) -> bool:
     "to_device",
 ]
 
-_all_ignore = ["sys", "math", "inspect", "warnings"]
-
 def __dir__() -> list[str]:
     return __all__
diff --git a/array_api_compat/common/_linalg.py b/array_api_compat/common/_linalg.py
@@ -225,8 +225,6 @@ def trace(
            'matrix_transpose', 'svdvals', 'vecdot', 'vector_norm', 'diagonal',
            'trace']
 
-_all_ignore = ['math', 'normalize_axis_tuple', 'get_xp', 'np', 'isdtype']
-
 
 def __dir__() -> list[str]:
     return __all__
diff --git a/array_api_compat/cupy/_aliases.py b/array_api_compat/cupy/_aliases.py
@@ -160,5 +160,3 @@ def count_nonzero(
                               'atan2', 'atanh', 'bitwise_left_shift',
                               'bitwise_invert', 'bitwise_right_shift',
                               'bool', 'concat', 'count_nonzero', 'pow', 'sign']
-
-_all_ignore = ['cp', 'get_xp']
diff --git a/array_api_compat/cupy/_typing.py b/array_api_compat/cupy/_typing.py
@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 __all__ = ["Array", "DType", "Device"]
-_all_ignore = ["cp"]
 
 from typing import TYPE_CHECKING
 
diff --git a/array_api_compat/dask/array/_aliases.py b/array_api_compat/dask/array/_aliases.py
@@ -369,8 +369,6 @@ def count_nonzero(
     "bitwise_left_shift", "bitwise_right_shift", "bitwise_invert",
 ]  # fmt: skip
 __all__ += _aliases.__all__
-_all_ignore = ["array_namespace", "get_xp", "da", "np"]
-
 
 def __dir__() -> list[str]:
     return __all__
diff --git a/array_api_compat/dask/array/fft.py b/array_api_compat/dask/array/fft.py
@@ -18,4 +18,6 @@
 rfftfreq = get_xp(da)(_fft.rfftfreq)
 
 __all__ = fft_all + ["fftfreq", "rfftfreq"]
-_all_ignore = ["da", "fft_all", "get_xp", "warnings"]
+
+def __dir__() -> list[str]:
+    return __all__
diff --git a/array_api_compat/dask/array/linalg.py b/array_api_compat/dask/array/linalg.py
@@ -69,4 +69,5 @@ def svdvals(x: _Array) -> _Array:
                         "cholesky", "matrix_rank", "matrix_norm", "svdvals",
                         "vector_norm", "diagonal"]
 
-_all_ignore = ['get_xp', 'da', 'linalg_all', 'warnings']
+def __dir__() -> list[str]:
+    return __all__
diff --git a/array_api_compat/numpy/_aliases.py b/array_api_compat/numpy/_aliases.py
@@ -157,7 +157,7 @@ def count_nonzero(
 else:
     unstack = get_xp(np)(_aliases.unstack)
 
-__all__ = [
+__all__ = _aliases.__all__ + [
     "__array_namespace_info__",
     "asarray",
     "astype",
@@ -176,8 +176,6 @@ def count_nonzero(
     "count_nonzero",
     "pow",
 ]
-__all__ += _aliases.__all__
-_all_ignore = ["np", "get_xp"]
 
 
 def __dir__() -> list[str]:
diff --git a/array_api_compat/numpy/_typing.py b/array_api_compat/numpy/_typing.py
@@ -23,7 +23,6 @@
     Array: TypeAlias = np.ndarray
 
 __all__ = ["Array", "DType", "Device"]
-_all_ignore = ["np"]
 
 
 def __dir__() -> list[str]:
diff --git a/array_api_compat/numpy/fft.py b/array_api_compat/numpy/fft.py
@@ -1,5 +1,4 @@
 import numpy as np
-from numpy.fft import __all__ as fft_all
 from numpy.fft import fft2, ifft2, irfft2, rfft2
 
 from .._internal import get_xp
@@ -21,15 +20,7 @@
 ifftshift = get_xp(np)(_fft.ifftshift)
 
 
-__all__ = ["rfft2", "irfft2", "fft2", "ifft2"]
-__all__ += _fft.__all__
-
+__all__ = _fft.__all__ + ["rfft2", "irfft2", "fft2", "ifft2"]
 
 def __dir__() -> list[str]:
     return __all__
-
-
-del get_xp
-del np
-del fft_all
-del _fft
diff --git a/array_api_compat/numpy/linalg.py b/array_api_compat/numpy/linalg.py
@@ -120,7 +120,7 @@ def solve(x1: Array, x2: Array, /) -> Array:
     vector_norm = get_xp(np)(_linalg.vector_norm)
 
 
-__all__ = [
+__all__ = _linalg.__all__ + [
     "LinAlgError",
     "cond",
     "det",
@@ -132,12 +132,11 @@ def solve(x1: Array, x2: Array, /) -> Array:
     "matrix_power",
     "multi_dot",
     "norm",
+    "solve", 
     "tensorinv",
     "tensorsolve",
+    "vector_norm",
 ]
-__all__ += _linalg.__all__
-__all__ += ["solve", "vector_norm"]
-
 
 def __dir__() -> list[str]:
     return __all__
diff --git a/array_api_compat/torch/_aliases.py b/array_api_compat/torch/_aliases.py
@@ -841,5 +841,3 @@ def sign(x: Array, /) -> Array:
            'unique_all', 'unique_counts', 'unique_inverse', 'unique_values',
            'matmul', 'matrix_transpose', 'vecdot', 'tensordot', 'isdtype',
            'take', 'take_along_axis', 'sign', 'finfo', 'iinfo', 'repeat']
-
-_all_ignore = ['torch', 'get_xp']
diff --git a/array_api_compat/torch/fft.py b/array_api_compat/torch/fft.py
@@ -81,5 +81,3 @@ def ifftshift(
     "fftshift",
     "ifftshift",
 ]
-
-_all_ignore = ['torch']
diff --git a/array_api_compat/torch/linalg.py b/array_api_compat/torch/linalg.py
@@ -113,9 +113,5 @@ def vector_norm(
 __all__ = linalg_all + ['outer', 'matmul', 'matrix_transpose', 'tensordot',
                         'cross', 'vecdot', 'solve', 'trace', 'vector_norm']
 
-_all_ignore = ['torch_linalg', 'sum']
-
-del linalg_all
-
 def __dir__() -> list[str]:
     return __all__
diff --git a/tests/test_all.py b/tests/test_all.py

Original file line number	Diff line number	Diff line change
`@@ -720,8 +720,6 @@ def iinfo(type_: DType \| Array, /, xp: Namespace) -> Any:`
`720`	`720`	`"finfo",`
`721`	`721`	`"iinfo",`
`722`	`722`	`]`
`723`		`-_all_ignore = ["inspect", "array_namespace", "NamedTuple"]`
`724`		`-`
`725`	`723`
`726`	`724`	`def __dir__() -> list[str]:`
`727`	`725`	`return __all__`
Original file line number	Diff line number	Diff line change
`@@ -1042,7 +1042,5 @@ def is_lazy_array(x: object) -> bool:`
`1042`	`1042`	`"to_device",`
`1043`	`1043`	`]`
`1044`	`1044`
`1045`		`-_all_ignore = ["sys", "math", "inspect", "warnings"]`
`1046`		`-`
`1047`	`1045`	`def __dir__() -> list[str]:`
`1048`	`1046`	`return __all__`