[ndarray][doc] Improve behaviors of 0darray + more docstring

numojo/core/_math_funcs.mojo

@@ -14,6 +14,7 @@ from memory import UnsafePointer
 
 from numojo.core.traits.backend import Backend
 from numojo.core.ndarray import NDArray
+from numojo.routines.creation import _0darray
 
 # TODO Add string method to give name
 
@@ -143,6 +144,13 @@ struct Vectorized(Backend):
         Returns:
             A a new NDArray that is NDArray with the function func applied.
         """
+
+        # For 0darray (numojo scalar)
+        # Treat it as a scalar and apply the function
+        if array.ndim == 0:
+            var result_array = _0darray(val=func[dtype, 1](array._buf.ptr[]))
+            return result_array
+
         var result_array: NDArray[dtype] = NDArray[dtype](array.shape)
         alias width = simdwidthof[dtype]()
 
@@ -186,6 +194,13 @@ struct Vectorized(Backend):
                 "Shape Mismatch error shapes must match for this function"
             )
 
+        # For 0darray (numojo scalar)
+        # Treat it as a scalar and apply the function
+        if array2.ndim == 0:
+            return self.math_func_1_array_1_scalar_in_one_array_out[
+                dtype, func
+            ](array1, array2[])
+
         var result_array: NDArray[dtype] = NDArray[dtype](array1.shape)
         alias width = simdwidthof[dtype]()
 
@@ -223,6 +238,12 @@ struct Vectorized(Backend):
             A a new NDArray that is NDArray with the function func applied.
         """
 
+        # For 0darray (numojo scalar)
+        # Treat it as a scalar and apply the function
+        if array.ndim == 0:
+            var result_array = _0darray(val=func[dtype, 1](array[], scalar))
+            return result_array
+
         var result_array: NDArray[dtype] = NDArray[dtype](array.shape)
         alias width = simdwidthof[dtype]()
 
@@ -249,6 +270,14 @@ struct Vectorized(Backend):
             raise Error(
                 "Shape Mismatch error shapes must match for this function"
             )
+
+        # For 0darray (numojo scalar)
+        # Treat it as a scalar and apply the function
+        if array2.ndim == 0:
+            return self.math_func_compare_array_and_scalar[dtype, func](
+                array1, array2[]
+            )
+
         var result_array: NDArray[DType.bool] = NDArray[DType.bool](
             array1.shape
         )
@@ -279,6 +308,12 @@ struct Vectorized(Backend):
     ](
         self: Self, array1: NDArray[dtype], scalar: SIMD[dtype, 1]
     ) raises -> NDArray[DType.bool]:
+        # For 0darray (numojo scalar)
+        # Treat it as a scalar and apply the function
+        if array1.ndim == 0:
+            var result_array = _0darray(val=func[dtype, 1](array1[], scalar))
+            return result_array
+
         var result_array: NDArray[DType.bool] = NDArray[DType.bool](
             array1.shape
         )

numojo/core/datatypes.mojo

@@ -235,5 +235,9 @@ fn _concise_dtype_str(dtype: DType) -> String:
         return "f32"
     elif dtype == f64:
         return "f64"
+    elif dtype == boolean:
+        return "boolean"
+    elif dtype == isize:
+        return "isize"
     else:
         return "Unknown"

numojo/core/item.mojo

@@ -133,9 +133,10 @@ struct Item(CollectionElement):
 
         if normalized_idx < 0 or normalized_idx >= self.ndim:
             raise Error(
-                String("Index ({}) out of range [{}, {})").format(
-                    index(idx), -self.ndim, self.ndim - 1
-                )
+                String(
+                    "Error in `numojo.Item.__getitem__()`: \n"
+                    "Index ({}) out of range [{}, {})\n"
+                ).format(index(idx), -self.ndim, self.ndim - 1)
             )
 
         return self._buf[normalized_idx]
@@ -159,9 +160,10 @@ struct Item(CollectionElement):
 
         if normalized_idx < 0 or normalized_idx >= self.ndim:
             raise Error(
-                String("Index ({}) out of range [{}, {})").format(
-                    index(idx), -self.ndim, self.ndim - 1
-                )
+                String(
+                    "Error in `numojo.Item.__getitem__()`: \n"
+                    "Index ({}) out of range [{}, {})\n"
+                ).format(index(idx), -self.ndim, self.ndim - 1)
             )
 
         self._buf[normalized_idx] = index(val)