API: Update lib.type_check namespace

mtsokol · mtsokol · commit 2f11e4987bda · 2023-08-25T12:12:37.000+02:00
diff --git a/numpy/__init__.py b/numpy/__init__.py
@@ -189,36 +189,40 @@
     from . import lib
     from .lib import (
         DataSource, angle, append, apply_along_axis, apply_over_axes,
-        array_split, asarray_chkfinite, asfarray, average, bartlett,
+        array_split, asarray_chkfinite, average, bartlett,
         bincount, blackman, broadcast_arrays, broadcast_shapes,
-        broadcast_to, byte_bounds, c_, column_stack, common_type,
+        broadcast_to, byte_bounds, c_, column_stack,
         copy, corrcoef, cov, delete, diag, diag_indices,
         diag_indices_from, diagflat, diff, digitize, dsplit, dstack,
         ediff1d, emath, expand_dims, extract, eye, fill_diagonal, fix,
         flip, fliplr, flipud, fromregex, get_array_wrap, genfromtxt,
         get_include, gradient, hamming, hanning, histogram2d,
-        hsplit, i0, imag, in1d,
-        index_exp, info, insert, interp, intersect1d, iscomplex,
-        iscomplexobj, isin, isneginf, isreal, isrealobj, issubclass_,
+        hsplit, i0, in1d,
+        index_exp, info, insert, interp, intersect1d,
+        isin, isneginf, issubclass_,
         issubsctype, iterable, ix_, kaiser, kron, load, loadtxt, mask_indices,
-        median, meshgrid, mgrid, mintypecode, nan_to_num, 
+        median, meshgrid, mgrid, 
         nanargmax, nanargmin, nancumprod, nancumsum, nanmax, nanmean,
         nanmedian, nanmin, nanpercentile, nanprod, nanquantile, nanstd,
         nansum, nanvar, ndenumerate, ndindex, ogrid, packbits, pad,
         percentile, piecewise, place, poly, poly1d, polyadd, polyder,
         polydiv, polyfit, polyint, polymul, polysub, polyval,
-        put_along_axis, quantile, r_, ravel_multi_index, real, real_if_close,
+        put_along_axis, quantile, r_, ravel_multi_index,
         roots, rot90, row_stack, s_, save, savetxt, savez, savez_compressed,
         select, setdiff1d, setxor1d, show_runtime, sinc, sort_complex, split,
         take_along_axis, tile, tracemalloc_domain, trapz, tri, tril,
-        tril_indices, tril_indices_from, typename, union1d, unique, unpackbits,
+        tril_indices, tril_indices_from, union1d, unique, unpackbits,
         unravel_index, unwrap, vander, vectorize, vsplit, trim_zeros,
         triu, triu_indices, triu_indices_from, isposinf, disp,
         deprecate, deprecate_with_doc, safe_eval, recfromtxt, recfromcsv
     )
     from .lib._histograms_impl import (
         histogram, histogram_bin_edges, histogramdd
     )
+    from .lib._type_check_impl import (
+        asfarray, iscomplexobj, isrealobj, imag, iscomplex, isreal,
+        nan_to_num, real, real_if_close, typename, mintypecode, common_type
+    )
     from . import matrixlib as _mat
     from .matrixlib import (
         asmatrix, bmat, mat, matrix
@@ -279,7 +283,8 @@
 
     __all__ = list(
         __numpy_submodules__ | set(core.__all__) | set(lib.__all__) | 
-        set(_mat.__all__) | set(lib._histograms_impl.__all__) | 
+        set(_mat.__all__) | set(lib._histograms_impl.__all__) |
+        set(lib._type_check_impl.__all__) |
         {"show_config", "__version__"}
     )
 
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi
@@ -578,7 +578,7 @@ from numpy.lib.twodim_base import (
     triu_indices_from as triu_indices_from,
 )
 
-from numpy.lib.type_check import (
+from numpy.lib._type_check_impl import (
     mintypecode as mintypecode,
     asfarray as asfarray,
     real as real,
diff --git a/numpy/lib/__init__.py b/numpy/lib/__init__.py
@@ -19,7 +19,7 @@
 
 # Private submodules
 # load module names. See https://github.com/networkx/networkx/issues/5838
-from . import type_check
+from . import _type_check_impl
 from . import index_tricks
 from . import function_base
 from . import nanfunctions
@@ -36,7 +36,6 @@
 from . import arraypad
 from . import _version
 
-from .type_check import *
 from .index_tricks import *
 from .function_base import *
 from .nanfunctions import *
@@ -57,7 +56,6 @@
 
 
 __all__ = ['emath', 'tracemalloc_domain']
-__all__ += type_check.__all__
 __all__ += index_tricks.__all__
 __all__ += function_base.__all__
 __all__ += shape_base.__all__
diff --git a/numpy/lib/__init__.pyi b/numpy/lib/__init__.pyi
@@ -188,21 +188,6 @@ from numpy.lib.twodim_base import (
     triu_indices_from as triu_indices_from,
 )
 
-from numpy.lib.type_check import (
-    mintypecode as mintypecode,
-    asfarray as asfarray,
-    real as real,
-    imag as imag,
-    iscomplex as iscomplex,
-    isreal as isreal,
-    iscomplexobj as iscomplexobj,
-    isrealobj as isrealobj,
-    nan_to_num as nan_to_num,
-    real_if_close as real_if_close,
-    typename as typename,
-    common_type as common_type,
-)
-
 from numpy.lib.ufunclike import (
     fix as fix,
     isposinf as isposinf,
diff --git a/numpy/lib/_type_check_impl.py b/numpy/lib/_type_check_impl.py
diff --git a/numpy/lib/_type_check_impl.pyi b/numpy/lib/_type_check_impl.pyi
diff --git a/numpy/lib/polynomial.py b/numpy/lib/polynomial.py
@@ -19,7 +19,7 @@
 from numpy.exceptions import RankWarning
 from numpy.lib.twodim_base import diag, vander
 from numpy.lib.function_base import trim_zeros
-from numpy.lib.type_check import iscomplex, real, imag, mintypecode
+from numpy.lib._type_check_impl import iscomplex, real, imag, mintypecode
 from numpy.linalg import eigvals, lstsq, inv
 
 
diff --git a/numpy/lib/scimath.py b/numpy/lib/scimath.py
@@ -34,7 +34,7 @@
 import numpy.core.numerictypes as nt
 from numpy.core.numeric import asarray, any
 from numpy.core.overrides import array_function_dispatch
-from numpy.lib.type_check import isreal
+from numpy.lib._type_check_impl import isreal
 
 
 __all__ = [
diff --git a/numpy/lib/tests/test_type_check.py b/numpy/lib/tests/test_type_check.py
@@ -1,11 +1,11 @@
 import numpy as np
-from numpy.testing import (
-    assert_, assert_equal, assert_array_equal, assert_raises
-    )
-from numpy.lib.type_check import (
+from numpy import (
     common_type, mintypecode, isreal, iscomplex, isposinf, isneginf,
     nan_to_num, isrealobj, iscomplexobj, asfarray, real_if_close
     )
+from numpy.testing import (
+    assert_, assert_equal, assert_array_equal, assert_raises
+    )
 
 
 def assert_all(x):
diff --git a/numpy/testing/_private/utils.py b/numpy/testing/_private/utils.py
@@ -277,7 +277,7 @@ def assert_equal(actual, desired, err_msg='', verbose=True):
                          verbose)
         return
     from numpy.core import ndarray, isscalar, signbit
-    from numpy.lib import iscomplexobj, real, imag
+    from numpy import iscomplexobj, real, imag
     if isinstance(actual, ndarray) or isinstance(desired, ndarray):
         return assert_array_equal(actual, desired, err_msg, verbose)
     msg = build_err_msg([actual, desired], err_msg, verbose=verbose)
@@ -482,7 +482,7 @@ def assert_almost_equal(actual, desired, decimal=7, err_msg='', verbose=True):
     """
     __tracebackhide__ = True  # Hide traceback for py.test
     from numpy.core import ndarray
-    from numpy.lib import iscomplexobj, real, imag
+    from numpy import iscomplexobj, real, imag
 
     # Handle complex numbers: separate into real/imag to handle
     # nan/inf/negative zero correctly
diff --git a/numpy/tests/test_public_api.py b/numpy/tests/test_public_api.py
@@ -213,7 +213,6 @@ def test_NPY_NO_EXPORT():
     "lib.polynomial",
     "lib.shape_base",
     "lib.twodim_base",
-    "lib.type_check",
     "lib.ufunclike",
     "lib.user_array",  # note: not in np.lib, but probably should just be deleted
     "lib.utils",

Original file line number	Diff line number	Diff line change
`@@ -578,7 +578,7 @@ from numpy.lib.twodim_base import (`
`578`	`578`	`triu_indices_from as triu_indices_from,`
`579`	`579`	`)`
`580`	`580`
`581`		`-from numpy.lib.type_check import (`
	`581`	`+from numpy.lib._type_check_impl import (`
`582`	`582`	`mintypecode as mintypecode,`
`583`	`583`	`asfarray as asfarray,`
`584`	`584`	`real as real,`