added returned type to array creation functions (issue larray-project#864)

Author: alixdamman

larray/core/array.py

@@ -8384,7 +8384,7 @@ def wrapper(*args, **kwargs):
 
 
 @_check_axes_argument
-def zeros(axes, title=None, dtype=float, order='C', meta=None):
+def zeros(axes, title=None, dtype=float, order='C', meta=None) -> Array:
     r"""Returns an array with the specified axes and filled with zeros.
 
     Parameters
@@ -8430,7 +8430,7 @@ def zeros(axes, title=None, dtype=float, order='C', meta=None):
     return Array(np.zeros(axes.shape, dtype, order), axes, meta=meta)
 
 
-def zeros_like(array, title=None, dtype=None, order='K', meta=None):
+def zeros_like(array, title=None, dtype=None, order='K', meta=None) -> Array:
     r"""Returns an array with the same axes as array and filled with zeros.
 
     Parameters
@@ -8466,7 +8466,7 @@ def zeros_like(array, title=None, dtype=None, order='K', meta=None):
 
 
 @_check_axes_argument
-def ones(axes, title=None, dtype=float, order='C', meta=None):
+def ones(axes, title=None, dtype=float, order='C', meta=None) -> Array:
     r"""Returns an array with the specified axes and filled with ones.
 
     Parameters
@@ -8502,7 +8502,7 @@ def ones(axes, title=None, dtype=float, order='C', meta=None):
     return Array(np.ones(axes.shape, dtype, order), axes, meta=meta)
 
 
-def ones_like(array, title=None, dtype=None, order='K', meta=None):
+def ones_like(array, title=None, dtype=None, order='K', meta=None) -> Array:
     r"""Returns an array with the same axes as array and filled with ones.
 
     Parameters
@@ -8539,7 +8539,7 @@ def ones_like(array, title=None, dtype=None, order='K', meta=None):
 
 
 @_check_axes_argument
-def empty(axes, title=None, dtype=float, order='C', meta=None):
+def empty(axes, title=None, dtype=float, order='C', meta=None) -> Array:
     r"""Returns an array with the specified axes and uninitialized (arbitrary) data.
 
     Parameters
@@ -8575,7 +8575,7 @@ def empty(axes, title=None, dtype=float, order='C', meta=None):
     return Array(np.empty(axes.shape, dtype, order), axes, meta=meta)
 
 
-def empty_like(array, title=None, dtype=None, order='K', meta=None):
+def empty_like(array, title=None, dtype=None, order='K', meta=None) -> Array:
     r"""Returns an array with the same axes as array and uninitialized (arbitrary) data.
 
     Parameters
@@ -8613,7 +8613,7 @@ def empty_like(array, title=None, dtype=None, order='K', meta=None):
 
 
 # We cannot use @_check_axes_argument here because an integer fill_value would be considered as an error
-def full(axes, fill_value, title=None, dtype=None, order='C', meta=None):
+def full(axes, fill_value, title=None, dtype=None, order='C', meta=None) -> Array:
     r"""Returns an array with the specified axes and filled with fill_value.
 
     Parameters
@@ -8665,7 +8665,7 @@ def full(axes, fill_value, title=None, dtype=None, order='C', meta=None):
     return res
 
 
-def full_like(array, fill_value, title=None, dtype=None, order='K', meta=None):
+def full_like(array, fill_value, title=None, dtype=None, order='K', meta=None) -> Array:
     r"""Returns an array with the same axes and type as input array and filled with fill_value.
 
     Parameters
@@ -8710,7 +8710,7 @@ def full_like(array, fill_value, title=None, dtype=None, order='K', meta=None):
 
 
 # XXX: would it be possible to generalize to multiple axes?
-def sequence(axis, initial=0, inc=None, mult=None, func=None, axes=None, title=None, meta=None):
+def sequence(axis, initial=0, inc=None, mult=None, func=None, axes=None, title=None, meta=None) -> Array:
     r"""
     Creates an array by sequentially applying modifications to the array along axis.
 
@@ -8993,7 +8993,7 @@ def ndrange(axes, start=0, title=None, dtype=int):
 
 
 @_check_axes_argument
-def ndtest(shape_or_axes, start=0, label_start=0, title=None, dtype=int, meta=None):
+def ndtest(shape_or_axes, start=0, label_start=0, title=None, dtype=int, meta=None) -> Array:
     r"""Returns test array with given shape.
 
     Axes are named by single letters starting from 'a'.
@@ -9093,7 +9093,7 @@ def kth_diag_indices(shape, k):
         return indices
 
 
-def diag(a, k=0, axes=(0, 1), ndim=2, split=True):
+def diag(a, k=0, axes=(0, 1), ndim=2, split=True) -> Array:
     r"""
     Extracts a diagonal or construct a diagonal array.
 
@@ -9175,7 +9175,7 @@ def diag(a, k=0, axes=(0, 1), ndim=2, split=True):
 
 
 @_check_axes_argument
-def labels_array(axes, title=None, meta=None):
+def labels_array(axes, title=None, meta=None) -> Array:
     r"""Returns an array with specified axes and the combination of
     corresponding labels as values.
 
@@ -9232,7 +9232,7 @@ def identity(axis):
                               "In other cases, you should use labels_array(axis) instead.")
 
 
-def eye(rows, columns=None, k=0, title=None, dtype=None, meta=None):
+def eye(rows, columns=None, k=0, title=None, dtype=None, meta=None) -> Array:
     r"""Returns a 2-D array with ones on the diagonal and zeros elsewhere.
 
     Parameters