Array Creation Routines: zeros and ones (#33)

* add functions for zeros and ones

too few args

formatting

new line

* black linting

Author: szb0

numpy-stubs/__init__.pyi

@@ -479,6 +479,12 @@ def array(
     subok: bool = ...,
     ndmin: int = ...,
 ) -> ndarray: ...
+def zeros(
+    shape: _ShapeLike, dtype: _DtypeLike = ..., order: Optional[str] = ...
+) -> ndarray: ...
+def ones(
+    shape: _ShapeLike, dtype: _DtypeLike = ..., order: Optional[str] = ...
+) -> ndarray: ...
 
 # TODO(shoyer): remove when the full numpy namespace is defined
 def __getattr__(name: str) -> Any: ...

tests/fail/simple.py

@@ -0,0 +1,10 @@
+"""Simple expression that should fail with mypy."""
+
+import numpy as np
+
+# Array creation routines checks
+np.zeros("test")  # E: incompatible type
+np.zeros()  # E: Too few arguments
+
+np.ones("test")  # E: incompatible type
+np.ones()  # E: Too few arguments

tests/pass/simple.py

@@ -13,6 +13,10 @@ def ndarray_func(x):
 array == 1
 array.dtype == float
 
+# Array creation routines checks
+ndarray_func(np.zeros([1, 2]))
+ndarray_func(np.ones([1, 2]))
+
 # Dtype construction
 np.dtype(float)
 np.dtype(np.float64)