STY: Partially apply ruff/pycodestyle rule E251

Unexpected spaces around keyword / parameter equals

For now, do not modify large and consistent code blocks that use spaces on
purpose, mostly when the default values are arrays.

Author: DimitriPapadopoulos

benchmarks/benchmarks/bench_ufunc.py

@@ -342,7 +342,7 @@ def time_ufunc_small_array(self, ufuncname):
         self.f(self.array_5)
 
     def time_ufunc_small_array_inplace(self, ufuncname):
-        self.f(self.array_5, out = self.array_5)
+        self.f(self.array_5, out=self.array_5)
 
     def time_ufunc_small_int_array(self, ufuncname):
         self.f(self.array_int_3)

numpy/_core/code_generators/generate_umath.py

@@ -1570,7 +1570,7 @@ def make_ufuncs(funcdict):
                 typenum=f"NPY_{english_upper(chartoname[c])}",
                 count=uf.nin + uf.nout,
                 name=name,
-                funcname = f"{english_upper(chartoname[c])}_{name}_indexed",
+                funcname=f"{english_upper(chartoname[c])}_{name}_indexed",
             ))
 
         mlist.append(r"""PyDict_SetItemString(dictionary, "%s", f);""" % name)

numpy/_core/tests/test_ufunc.py

@@ -2104,7 +2104,7 @@ def __rmul__(self, other):
     def test_array_wrap_array_priority(self):
         class ArrayPriorityBase(np.ndarray):
             @classmethod
-            def __array_wrap__(cls, array, context=None, return_scalar = False):
+            def __array_wrap__(cls, array, context=None, return_scalar=False):
                 return cls
 
         class ArrayPriorityMinus0(ArrayPriorityBase):

numpy/_core/tests/test_umath_accuracy.py

@@ -75,7 +75,7 @@ def test_validate_transcendentals(self):
                         assert_array_max_ulp(npfunc(inval), outval, maxulperr)
 
     @pytest.mark.skipif(IS_AVX512FP16,
-            reason = "SVML FP16 have slightly higher ULP errors")
+            reason="SVML FP16 have slightly higher ULP errors")
     @pytest.mark.parametrize("ufunc", UNARY_OBJECT_UFUNCS)
     def test_validate_fp16_transcendentals(self, ufunc):
         with np.errstate(all='ignore'):

numpy/f2py/auxfuncs.py

@@ -923,7 +923,7 @@ def getuseblocks(pymod):
                 all_uses.extend([x for x in modblock.get("use").keys() if "__" not in x])
     return all_uses
 
-def process_f2cmap_dict(f2cmap_all, new_map, c2py_map, verbose = False):
+def process_f2cmap_dict(f2cmap_all, new_map, c2py_map, verbose=False):
     """
     Update the Fortran-to-C type mapping dictionary with new mappings and
     return a list of successfully mapped C types.

numpy/f2py/tests/test_array_from_pyobj.py

@@ -33,7 +33,7 @@ def setup_module():
         src = [
             get_testdir() / "wrapmodule.c",
         ]
-        wrap = util.build_meson(src, module_name = "test_array_from_pyobj_ext")
+        wrap = util.build_meson(src, module_name="test_array_from_pyobj_ext")
 
 
 def flags_info(arr):

numpy/f2py/tests/test_regression.py

@@ -53,7 +53,7 @@ def ubound(xl, xh):
             return xh - xl + 1
         rval = self.module.foo(is_=xlow, ie_=xhigh,
                         arr=xvec[:ubound(xlow, xhigh)])
-        expval = np.arange(11, dtype = np.float32)
+        expval = np.arange(11, dtype=np.float32)
         assert np.allclose(rval, expval)
 
 

numpy/lib/tests/test_function_base.py

@@ -1962,7 +1962,7 @@ def f(x):
 
     def test_bad_input(self):
         with assert_raises(TypeError):
-            A = np.vectorize(pyfunc = 3)
+            A = np.vectorize(pyfunc=3)
 
     def test_no_keywords(self):
         with assert_raises(TypeError):
@@ -3887,7 +3887,7 @@ def test_quantile_preserve_int_type(self, dtype):
     def test_q_zero_one(self, method):
         # gh-24710
         arr = [10, 11, 12]
-        quantile = np.quantile(arr, q = [0, 1], method=method)
+        quantile = np.quantile(arr, q=[0, 1], method=method)
         assert_equal(quantile, np.array([10, 12]))
 
     @pytest.mark.parametrize("method", quantile_methods)
@@ -4163,10 +4163,10 @@ class TestLerp:
                                    min_value=0, max_value=1),
                       t1=st.floats(allow_nan=False, allow_infinity=False,
                                    min_value=0, max_value=1),
-                      a = st.floats(allow_nan=False, allow_infinity=False,
-                                    min_value=-1e300, max_value=1e300),
-                      b = st.floats(allow_nan=False, allow_infinity=False,
-                                    min_value=-1e300, max_value=1e300))
+                      a=st.floats(allow_nan=False, allow_infinity=False,
+                                  min_value=-1e300, max_value=1e300),
+                      b=st.floats(allow_nan=False, allow_infinity=False,
+                                  min_value=-1e300, max_value=1e300))
     def test_linear_interpolation_formula_monotonic(self, t0, t1, a, b):
         l0 = nfb._lerp(a, b, t0)
         l1 = nfb._lerp(a, b, t1)

numpy/lib/tests/test_histograms.py

@@ -588,7 +588,7 @@ def test_simple_range(self):
             x3 = np.linspace(-100, -50, testlen)
             x = np.hstack((x1, x2, x3))
             for estimator, numbins in expectedResults.items():
-                a, b = np.histogram(x, estimator, range = (-20, 20))
+                a, b = np.histogram(x, estimator, range=(-20, 20))
                 msg = f"For the {estimator} estimator"
                 msg += f" with datasize of {testlen}"
                 assert_equal(len(a), numbins, err_msg=msg)

numpy/ma/extras.pyi

@@ -56,7 +56,7 @@ __all__ = [
 ]
 
 def count_masked(arr, axis=...): ...
-def masked_all(shape, dtype = ...): ...
+def masked_all(shape, dtype=...): ...
 def masked_all_like(arr): ...
 
 class _fromnxfunction:
@@ -96,8 +96,8 @@ def compress_nd(x, axis=...): ...
 def compress_rowcols(x, axis=...): ...
 def compress_rows(a): ...
 def compress_cols(a): ...
-def mask_rows(a, axis = ...): ...
-def mask_cols(a, axis = ...): ...
+def mask_rows(a, axis=...): ...
+def mask_cols(a, axis=...): ...
 def ediff1d(arr, to_end=..., to_begin=...): ...
 def unique(ar1, return_index=..., return_inverse=...): ...
 def intersect1d(ar1, ar2, assume_unique=...): ...
@@ -107,7 +107,7 @@ def isin(element, test_elements, assume_unique=..., invert=...): ...
 def union1d(ar1, ar2): ...
 def setdiff1d(ar1, ar2, assume_unique=...): ...
 def cov(x, y=..., rowvar=..., bias=..., allow_masked=..., ddof=...): ...
-def corrcoef(x, y=..., rowvar=..., bias = ..., allow_masked=..., ddof = ...): ...
+def corrcoef(x, y=..., rowvar=..., bias=..., allow_masked=..., ddof=...): ...
 
 class MAxisConcatenator(AxisConcatenator):
     @staticmethod