mgrid ogrid classes (#366)

* mgrid, ogrid impl

Author: Alexander-Makaryev

dpnp/dpnp_iface_arraycreation.py

@@ -64,6 +64,8 @@
     "linspace",
     "logspace",
     "meshgrid",
+    "mgrid",
+    "ogrid",
     "ones",
     "ones_like",
     "tri",
@@ -771,6 +773,66 @@ def meshgrid(*xi, copy=True, sparse=False, indexing='xy'):
     return call_origin(numpy.meshgrid, xi, copy, sparse, indexing)
 
 
+class MGridClass:
+    """
+    Construct a dense multi-dimensional "meshgrid".
+
+    For full documentation refer to :obj:`numpy.mgrid`.
+
+    Examples
+    --------
+    >>> import dpnp as np
+    >>> np.mgrid[0:5,0:5]
+    array([[[0, 0, 0, 0, 0],
+            [1, 1, 1, 1, 1],
+            [2, 2, 2, 2, 2],
+            [3, 3, 3, 3, 3],
+            [4, 4, 4, 4, 4]],
+           [[0, 1, 2, 3, 4],
+            [0, 1, 2, 3, 4],
+            [0, 1, 2, 3, 4],
+            [0, 1, 2, 3, 4],
+            [0, 1, 2, 3, 4]]])
+    >>> np.mgrid[-1:1:5j]
+    array([-1. , -0.5,  0. ,  0.5,  1. ])
+
+    """
+
+    def __getitem__(self, key):
+        return dpnp.array(numpy.mgrid[key])
+
+
+mgrid = MGridClass()
+
+
+class OGridClass:
+    """
+    Construct an open multi-dimensional "meshgrid".
+
+    For full documentation refer to :obj:`numpy.ogrid`.
+
+    Examples
+    --------
+    >>> import dpnp as np
+    >>> from numpy import ogrid
+    >>> ogrid[-1:1:5j]
+    array([-1. , -0.5,  0. ,  0.5,  1. ])
+    >>> ogrid[0:5,0:5]
+    [array([[0],
+            [1],
+            [2],
+            [3],
+            [4]]), array([[0, 1, 2, 3, 4]])]
+
+    """
+
+    def __getitem__(self, key):
+        return dpnp.array(numpy.ogrid[key])
+
+
+ogrid = OGridClass()
+
+
 def ones(shape, dtype=None, order='C'):
     """
     Return a new array of given shape and type, filled with ones.

tests/skipped_tests.tbl

@@ -321,15 +321,7 @@ tests/third_party/cupy/creation_tests/test_ranges.py::TestRanges::test_linspace_
 tests/third_party/cupy/creation_tests/test_ranges.py::TestRanges::test_linspace_mixed_start_stop
 tests/third_party/cupy/creation_tests/test_ranges.py::TestRanges::test_linspace_mixed_start_stop2
 tests/third_party/cupy/creation_tests/test_ranges.py::TestRanges::test_linspace_start_stop_list
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid0
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid1
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid2
 tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid3
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid4
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid5
-tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid0
-tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid1
-tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid2
 tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid3
 tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid4
 tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid5

tests/skipped_tests_gpu.tbl

@@ -373,15 +373,7 @@ tests/third_party/cupy/creation_tests/test_ranges.py::TestMeshgrid_param_7_{copy
 tests/third_party/cupy/creation_tests/test_ranges.py::TestMeshgrid_param_7_{copy=True, indexing='ij', sparse=True}::test_meshgrid1
 tests/third_party/cupy/creation_tests/test_ranges.py::TestMeshgrid_param_7_{copy=True, indexing='ij', sparse=True}::test_meshgrid2
 tests/third_party/cupy/creation_tests/test_ranges.py::TestMeshgrid_param_7_{copy=True, indexing='ij', sparse=True}::test_meshgrid3
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid0
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid1
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid2
 tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid3
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid4
-tests/third_party/cupy/creation_tests/test_ranges.py::TestMgrid::test_mgrid5
-tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid0
-tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid1
-tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid2
 tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid3
 tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid4
 tests/third_party/cupy/creation_tests/test_ranges.py::TestOgrid::test_ogrid5