add numpy.asscalar() (#22947)

ivy-llc · Sep 4, 2023 · 2ec34e2 · 2ec34e2
1 parent e1f93d2
commit 2ec34e2
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/ivy/functional/frontends/numpy/manipulation_routines/changing_kind_of_array.py b/ivy/functional/frontends/numpy/manipulation_routines/changing_kind_of_array.py
@@ -5,3 +5,7 @@
 
 def asmatrix(data, dtype=None):
     return np_frontend.matrix(ivy.array(data), dtype=dtype, copy=False)
+
+
+def asscalar(a):
+    return a.item()
diff --git a/...t_ivy/test_frontends/test_numpy/test_manipulation_routines/test_changing_kind_of_array.py b/...t_ivy/test_frontends/test_numpy/test_manipulation_routines/test_changing_kind_of_array.py
@@ -18,3 +18,13 @@ def test_numpy_asmatrix(arr, backend_fw):
         ret_gt = np.asmatrix(x[0])
         assert ret.shape == ret_gt.shape
         assert ivy_backend.all(ivy_backend.flatten(ret._data) == np.ravel(ret_gt))
+
+
+@handle_frontend_test(
+    fn_tree="numpy.asscalar",
+    arr=helpers.array_values(dtype=helpers.get_dtypes("numeric"), shape=1),
+)
+def test_numpy_asscalar(arr: np.ndarray):
+    ret_1 = arr.item()
+    ret_2 = np_frontend.asscalar(arr)
+    assert ret_1 == ret_2