MAINT: run black on all files in CI (#64)

In #63 black was run on
`setup.py`. Follow up by running black on all files in CI. Do an
inital pass over the entire codebase so that the CI check will pass.

Author: person142

.travis.yml

@@ -17,7 +17,7 @@ install:
 
 script:
 - flake8
-- black --check numpy-stubs
+- black --check .
 - py.test
 
 cache:

runtests.py

@@ -10,12 +10,12 @@
 
 def main():
     subprocess.run(
-        [sys.executable, '-m', 'pip', 'install', STUBS_ROOT],
+        [sys.executable, "-m", "pip", "install", STUBS_ROOT],
         capture_output=True,
         check=True,
     )
     sys.exit(pytest.main([STUBS_ROOT] + sys.argv[1:]))
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()

scripts/find_ufuncs.py

@@ -10,12 +10,12 @@ def main():
 
     ufunc_stubs = []
     for ufunc in set(ufuncs):
-        ufunc_stubs.append(f'{ufunc.__name__}: ufunc')
+        ufunc_stubs.append(f"{ufunc.__name__}: ufunc")
     ufunc_stubs.sort()
 
     for stub in ufunc_stubs:
         print(stub)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()

tests/fail/scalars.py

@@ -28,8 +28,8 @@
 
 np.datetime64(0)  # E: non-matching overload
 
-dt_64 = np.datetime64(0, 'D')
-td_64 = np.timedelta64(1, 'h')
+dt_64 = np.datetime64(0, "D")
+td_64 = np.timedelta64(1, "h")
 
 dt_64 + dt_64  # E: Unsupported operand types
 

tests/fail/ufuncs.py

@@ -1,5 +1,5 @@
 import numpy as np
 
-np.sin.nin + 'foo'  # E: Unsupported operand types
-np.sin(1, foo='bar')  # E: Unexpected keyword argument
-np.sin(1, extobj=['foo', 'foo', 'foo'])  # E: incompatible type
+np.sin.nin + "foo"  # E: Unsupported operand types
+np.sin(1, foo="bar")  # E: Unexpected keyword argument
+np.sin(1, extobj=["foo", "foo", "foo"])  # E: incompatible type

tests/fail/warnings_and_errors.py

@@ -2,4 +2,6 @@
 
 np.AxisError(1.0)  # E: Argument 1 to "AxisError" has incompatible type
 np.AxisError(1, ndim=2.0)  # E: Argument "ndim" to "AxisError" has incompatible type
-np.AxisError(2, msg_prefix=404)  # E: Argument "msg_prefix" to "AxisError" has incompatible type
+np.AxisError(
+    2, msg_prefix=404  # E: Argument "msg_prefix" to "AxisError" has incompatible type
+)

tests/pass/scalars.py

@@ -53,23 +53,23 @@ def __float__(self):
 
 # Time structures
 np.datetime64()
-np.datetime64(0, 'D')
-np.datetime64('2019')
-np.datetime64('2019', 'D')
+np.datetime64(0, "D")
+np.datetime64("2019")
+np.datetime64("2019", "D")
 
 np.timedelta64()
 np.timedelta64(0)
-np.timedelta64(0, 'D')
+np.timedelta64(0, "D")
 
-dt_64 = np.datetime64(0, 'D')
-td_64 = np.timedelta64(1, 'h')
+dt_64 = np.datetime64(0, "D")
+td_64 = np.timedelta64(1, "h")
 
 dt_64 + td_64
 dt_64 - dt_64
 dt_64 - td_64
 
 td_64 + td_64
 td_64 - td_64
-td_64 / 1.
+td_64 / 1.0
 td_64 / td_64
 td_64 % td_64

tests/pass/simple.py

@@ -6,9 +6,13 @@
 
 # Basic checks
 array = np.array([1, 2])
+
+
 def ndarray_func(x):
     # type: (np.ndarray) -> np.ndarray
     return x
+
+
 ndarray_func(np.array([1, 2]))
 array == 1
 array.dtype == float
@@ -21,38 +25,40 @@ def ndarray_func(x):
 np.dtype(float)
 np.dtype(np.float64)
 np.dtype(None)
-np.dtype('float64')
+np.dtype("float64")
 np.dtype(np.dtype(float))
-np.dtype(('U', 10))
+np.dtype(("U", 10))
 np.dtype((np.int32, (2, 2)))
 # Define the arguments on the previous line to prevent bidirectional
 # type inference in mypy from broadening the types.
-two_tuples_dtype = [('R', 'u1'), ('G', 'u1'), ('B', 'u1')]
+two_tuples_dtype = [("R", "u1"), ("G", "u1"), ("B", "u1")]
 np.dtype(two_tuples_dtype)
 
-three_tuples_dtype = [('R', 'u1', 1)]
+three_tuples_dtype = [("R", "u1", 1)]
 np.dtype(three_tuples_dtype)
 
-mixed_tuples_dtype = [('R', 'u1'), ('G', np.unicode_, 1)]
+mixed_tuples_dtype = [("R", "u1"), ("G", np.unicode_, 1)]
 np.dtype(mixed_tuples_dtype)
 
-shape_tuple_dtype = [('R', 'u1', (2, 2))]
+shape_tuple_dtype = [("R", "u1", (2, 2))]
 np.dtype(shape_tuple_dtype)
 
-shape_like_dtype = [('R', 'u1', (2, 2)), ('G', np.unicode_, 1)]
+shape_like_dtype = [("R", "u1", (2, 2)), ("G", np.unicode_, 1)]
 np.dtype(shape_like_dtype)
 
-object_dtype = [('field1', object)]
+object_dtype = [("field1", object)]
 np.dtype(object_dtype)
 
-np.dtype({'col1': ('U10', 0), 'col2': ('float32', 10)})
-np.dtype((np.int32, {'real': (np.int16, 0), 'imag': (np.int16, 2)}))
+np.dtype({"col1": ("U10", 0), "col2": ("float32", 10)})
+np.dtype((np.int32, {"real": (np.int16, 0), "imag": (np.int16, 2)}))
 np.dtype((np.int32, (np.int8, 4)))
 
 # Iteration and indexing
 def iterable_func(x):
     # type: (Iterable) -> Iterable
     return x
+
+
 iterable_func(array)
 [element for element in array]
 iter(array)

tests/pass/ufuncs.py

@@ -3,12 +3,8 @@
 np.sin(1)
 np.sin([1, 2, 3])
 np.sin(1, out=np.empty(1))
-np.matmul(
-    np.ones((2, 2, 2)),
-    np.ones((2, 2, 2)),
-    axes=[(0, 1), (0, 1), (0, 1)],
-)
-np.sin(1, signature='D')
+np.matmul(np.ones((2, 2, 2)), np.ones((2, 2, 2)), axes=[(0, 1), (0, 1), (0, 1)])
+np.sin(1, signature="D")
 np.sin(1, extobj=[16, 1, lambda: None])
 np.sin(1) + np.sin(1)
 np.sin.types[0]

tests/pass/warnings_and_errors.py

@@ -3,5 +3,5 @@
 np.AxisError(1)
 np.AxisError(1, ndim=2)
 np.AxisError(1, ndim=None)
-np.AxisError(1, ndim=2, msg_prefix='error')
+np.AxisError(1, ndim=2, msg_prefix="error")
 np.AxisError(1, ndim=2, msg_prefix=None)