API: Update lib.index_tricks namespace

mtsokol · mtsokol · commit eab45a4402d7 · 2023-08-29T14:13:08.000+02:00
diff --git a/numpy/__init__.py b/numpy/__init__.py
@@ -185,21 +185,19 @@
     from .lib import (
         DataSource, apply_along_axis, apply_over_axes,
         array_split, broadcast_arrays, broadcast_shapes,
-        broadcast_to, byte_bounds, c_, column_stack, diag, diag_indices,
-        diag_indices_from, diagflat, dsplit, dstack,
-        ediff1d, emath, expand_dims, eye, fill_diagonal, fix,
+        broadcast_to, byte_bounds, column_stack, diag,
+        diagflat, dsplit, dstack, ediff1d, emath, expand_dims, eye, fix,
         fliplr, flipud, fromregex, get_array_wrap, genfromtxt,
-        get_include, histogram2d, hsplit, in1d, index_exp, info, intersect1d,
-        isin, isneginf, ix_, kron, load, loadtxt, mask_indices,
-        mgrid, ndenumerate, ndindex, ogrid, 
+        get_include, histogram2d, hsplit, in1d, info, intersect1d,
+        isin, isneginf, kron, load, loadtxt, mask_indices,
         packbits, pad, poly, poly1d, polyadd, polyder,
         polydiv, polyfit, polyint, polymul, polysub, polyval,
-        put_along_axis, r_, ravel_multi_index,
-        roots, row_stack, s_, save, savetxt, savez, savez_compressed,
+        put_along_axis,
+        roots, row_stack, save, savetxt, savez, savez_compressed,
         setdiff1d, setxor1d, show_runtime, split,
         take_along_axis, tile, tri, tril,
         tril_indices, tril_indices_from, union1d, unique, unpackbits,
-        unravel_index, vander, vsplit, triu, triu_indices, triu_indices_from, 
+        vander, vsplit, triu, triu_indices, triu_indices_from, 
         isposinf
     )
     from .lib._histograms_impl import (
@@ -221,6 +219,11 @@
         iscomplexobj, isrealobj, imag, iscomplex, isreal, nan_to_num, real, 
         real_if_close, typename, mintypecode, common_type
     )
+    from .lib._index_tricks_impl import (
+        diag_indices_from, diag_indices, fill_diagonal, ndindex, ndenumerate,
+        ix_, c_, r_, s_, ogrid, mgrid, unravel_index, ravel_multi_index, 
+        index_exp
+    )
     from . import matrixlib as _mat
     from .matrixlib import (
         asmatrix, bmat, matrix
@@ -285,6 +288,7 @@
         set(lib._nanfunctions_impl.__all__) |
         set(lib._function_base_impl.__all__) |
         set(lib._type_check_impl.__all__) |
+        set(lib._index_tricks_impl.__all__) |
         {"show_config", "__version__"}
     )
 
diff --git a/numpy/__init__.pyi b/numpy/__init__.pyi
@@ -458,7 +458,7 @@ from numpy.lib._histograms_impl import (
     histogramdd as histogramdd,
 )
 
-from numpy.lib.index_tricks import (
+from numpy.lib._index_tricks_impl import (
     ravel_multi_index as ravel_multi_index,
     unravel_index as unravel_index,
     mgrid as mgrid,
diff --git a/numpy/core/tests/test_function_base.py b/numpy/core/tests/test_function_base.py
@@ -460,7 +460,7 @@ def test_add_doc(self):
         tgt = "Current flat index into the array."
         assert_equal(np.core.flatiter.index.__doc__[:len(tgt)], tgt)
         assert_(len(np.core.ufunc.identity.__doc__) > 300)
-        assert_(len(np.lib.index_tricks.mgrid.__doc__) > 300)
+        assert_(len(np.lib._index_tricks_impl.mgrid.__doc__) > 300)
 
     @pytest.mark.skipif(sys.flags.optimize == 2, reason="Python running -OO")
     def test_errors_are_ignored(self):
diff --git a/numpy/lib/__init__.py b/numpy/lib/__init__.py
@@ -20,7 +20,7 @@
 # Private submodules
 # load module names. See https://github.com/networkx/networkx/issues/5838
 from . import _type_check_impl
-from . import index_tricks
+from . import _index_tricks_impl
 from . import _nanfunctions_impl
 from . import _function_base_impl
 from . import shape_base
@@ -36,7 +36,6 @@
 from . import arraypad
 from . import _version
 
-from .index_tricks import *
 from .shape_base import *
 from .stride_tricks import *
 from .twodim_base import *
@@ -54,7 +53,6 @@
 
 
 __all__ = ['emath']
-__all__ += index_tricks.__all__
 __all__ += shape_base.__all__
 __all__ += stride_tricks.__all__
 __all__ += twodim_base.__all__
diff --git a/numpy/lib/__init__.pyi b/numpy/lib/__init__.pyi
@@ -40,21 +40,6 @@ from numpy.lib.arrayterator import (
     Arrayterator as Arrayterator,
 )
 
-from numpy.lib.index_tricks import (
-    ravel_multi_index as ravel_multi_index,
-    unravel_index as unravel_index,
-    mgrid as mgrid,
-    ogrid as ogrid,
-    r_ as r_,
-    c_ as c_,
-    s_ as s_,
-    index_exp as index_exp,
-    ix_ as ix_,
-    fill_diagonal as fill_diagonal,
-    diag_indices as diag_indices,
-    diag_indices_from as diag_indices_from,
-)
-
 from numpy.lib.npyio import (
     savetxt as savetxt,
     loadtxt as loadtxt,
diff --git a/numpy/lib/_index_tricks_impl.py b/numpy/lib/_index_tricks_impl.py
diff --git a/numpy/lib/_index_tricks_impl.pyi b/numpy/lib/_index_tricks_impl.pyi
diff --git a/numpy/lib/arraypad.py b/numpy/lib/arraypad.py
@@ -5,7 +5,7 @@
 """
 import numpy as np
 from numpy.core.overrides import array_function_dispatch
-from numpy.lib.index_tricks import ndindex
+from numpy.lib._index_tricks_impl import ndindex
 
 
 __all__ = ['pad']
diff --git a/numpy/lib/shape_base.py b/numpy/lib/shape_base.py
@@ -9,7 +9,7 @@
 from numpy.core import vstack, atleast_3d
 from numpy.core.numeric import normalize_axis_tuple
 from numpy.core.shape_base import _arrays_for_stack_dispatcher
-from numpy.lib.index_tricks import ndindex
+from numpy.lib._index_tricks_impl import ndindex
 from numpy.matrixlib.defmatrix import matrix  # this raises all the right alarm bells
 
 
diff --git a/numpy/lib/tests/test_index_tricks.py b/numpy/lib/tests/test_index_tricks.py
@@ -5,9 +5,9 @@
     assert_, assert_equal, assert_array_equal, assert_almost_equal,
     assert_array_almost_equal, assert_raises, assert_raises_regex,
     )
-from numpy.lib.index_tricks import (
+from numpy.lib._index_tricks_impl import (
     mgrid, ogrid, ndenumerate, fill_diagonal, diag_indices, diag_indices_from,
-    index_exp, ndindex, r_, s_, ix_
+    index_exp, ndindex, c_, r_, s_, ix_
     )
 
 
@@ -405,7 +405,7 @@ def test_repeated_input(self):
 
 
 def test_c_():
-    a = np.c_[np.array([[1, 2, 3]]), 0, 0, np.array([[4, 5, 6]])]
+    a = c_[np.array([[1, 2, 3]]), 0, 0, np.array([[4, 5, 6]])]
     assert_equal(a, [[1, 2, 3, 0, 0, 4, 5, 6]])
 
 
diff --git a/numpy/ma/extras.py b/numpy/ma/extras.py
@@ -35,7 +35,7 @@
 from numpy.core.multiarray import normalize_axis_index
 from numpy.core.numeric import normalize_axis_tuple
 from numpy.lib._function_base_impl import _ureduce
-from numpy.lib.index_tricks import AxisConcatenator
+from numpy.lib._index_tricks_impl import AxisConcatenator
 
 
 def issequence(seq):
diff --git a/numpy/tests/test_public_api.py b/numpy/tests/test_public_api.py
@@ -199,7 +199,6 @@ def test_NPY_NO_EXPORT():
     "lib.arraypad",
     "lib.arraysetops",
     "lib.arrayterator",
-    "lib.index_tricks",
     "lib.npyio",
     "lib.polynomial",
     "lib.shape_base",

Original file line number	Diff line number	Diff line change
`@@ -458,7 +458,7 @@ from numpy.lib._histograms_impl import (`
`458`	`458`	`histogramdd as histogramdd,`
`459`	`459`	`)`
`460`	`460`
`461`		`-from numpy.lib.index_tricks import (`
	`461`	`+from numpy.lib._index_tricks_impl import (`
`462`	`462`	`ravel_multi_index as ravel_multi_index,`
`463`	`463`	`unravel_index as unravel_index,`
`464`	`464`	`mgrid as mgrid,`
Original file line number	Diff line number	Diff line change
`@@ -5,9 +5,9 @@`
`5`	`5`	`assert_, assert_equal, assert_array_equal, assert_almost_equal,`
`6`	`6`	`assert_array_almost_equal, assert_raises, assert_raises_regex,`
`7`	`7`	`)`
`8`		`-from numpy.lib.index_tricks import (`
	`8`	`+from numpy.lib._index_tricks_impl import (`
`9`	`9`	`mgrid, ogrid, ndenumerate, fill_diagonal, diag_indices, diag_indices_from,`
`10`		`- index_exp, ndindex, r_, s_, ix_`
	`10`	`+ index_exp, ndindex, c_, r_, s_, ix_`
`11`	`11`	`)`
`12`	`12`
`13`	`13`
`@@ -405,7 +405,7 @@ def test_repeated_input(self):`
`405`	`405`
`406`	`406`
`407`	`407`	`def test_c_():`
`408`		`- a = np.c_[np.array([[1, 2, 3]]), 0, 0, np.array([[4, 5, 6]])]`
	`408`	`+ a = c_[np.array([[1, 2, 3]]), 0, 0, np.array([[4, 5, 6]])]`
`409`	`409`	`assert_equal(a, [[1, 2, 3, 0, 0, 4, 5, 6]])`
`410`	`410`
`411`	`411`