MAINT: linting

mattip · mattip · commit 1db1ffe140b3 · 2023-02-01T16:22:26.000+02:00
diff --git a/numpy/core/tests/test_ufunc.py b/numpy/core/tests/test_ufunc.py
@@ -1988,12 +1988,14 @@ def test_ufunc_at_basic(self, a):
             np.add.at(a, [2, 5, 3], [[1, 2], 1])
     
     # ufuncs with indexed loops for perfomance in ufunc.at
-    indexed_ufuncs = [np.add, np.subtract, np.multiply, np.floor_divide, np.divide]
+    indexed_ufuncs = [np.add, np.subtract, np.multiply,
+                      np.floor_divide, np.divide]
+
     @pytest.mark.parametrize(
                 "typecode", np.typecodes['AllInteger'] + np.typecodes['Float'])
     @pytest.mark.parametrize("ufunc", indexed_ufuncs)
     def test_ufunc_at_inner_loops(self, typecode, ufunc):
-        if ufunc is np.divide and typecode in  np.typecodes['AllInteger']:
+        if ufunc is np.divide and typecode in np.typecodes['AllInteger']:
             # Avoid divide-by-zero and inf for integer divide
             a = np.ones(100, dtype=typecode)
             indx = np.random.randint(100, size=30, dtype=np.intp)
@@ -2009,7 +2011,7 @@ def test_ufunc_at_inner_loops(self, typecode, ufunc):
             ufunc.at(a, indx, vals)
         with warnings.catch_warnings(record=True) as w_loop:
             warnings.simplefilter('always')
-            for i,v in zip(indx, vals):
+            for i, v in zip(indx, vals):
                 # Make sure all the work happens inside the ufunc
                 # in order to duplicate error/warning handling
                 ufunc(atag[i], v, out=atag[i:i+1], casting="unsafe")
