reformat with black

Author: andrenarchy

krypy/deflation.py

@@ -213,9 +213,7 @@ def estimate_time(self, nsteps, ndefl, deflweight=1.0):
             "M": ndefl,
             "Ml": ndefl,
             "Mr": ndefl,
-            "ip_B": (
-                ndefl * (ndefl + 1) / 2 + ndefl ** 2 + 2 * ndefl * solver_ops["Ml"]
-            ),
+            "ip_B": (ndefl * (ndefl + 1) / 2 + ndefl**2 + 2 * ndefl * solver_ops["Ml"]),
             "axpy": (
                 ndefl * (ndefl + 1) / 2
                 + ndefl * ndefl
@@ -670,7 +668,7 @@ def _auto():
         )[0:-1]
 
         def compute_pseudo(delta_log):
-            delta = 10 ** delta_log
+            delta = 10**delta_log
             if ls_small.self_adjoint:
                 # pseudospectrum are intervals
                 pseudo_intervals = utils.Intervals(

krypy/linsys.py

@@ -510,8 +510,7 @@ def operations(nsteps):
         )
 
     def _solve(self):
-        """Abstract method that solves the linear system.
-        """
+        """Abstract method that solves the linear system."""
         raise NotImplementedError(
             "_solve has to be overridden by " "the derived solver class."
         )
@@ -597,7 +596,7 @@ def _solve(self):
         yk = numpy.zeros((N, 1), dtype=self.dtype)
 
         # square of the old residual norm
-        self.rhos = rhos = [self.MMlr0_norm ** 2]
+        self.rhos = rhos = [self.MMlr0_norm**2]
 
         # will be updated by _compute_rkn if explicit_residual is True
         self.Mlrk = self.Mlr0.copy()
@@ -662,7 +661,7 @@ def _solve(self):
 
             # compute norm and rho_new
             MMlrk_norm = utils.norm(self.Mlrk, self.MMlrk, ip_B=self.linear_system.ip_B)
-            rhos.append(MMlrk_norm ** 2)
+            rhos.append(MMlrk_norm**2)
 
             # compute Lanczos vector + new subdiagonal element
             if self.store_arnoldi:
@@ -680,7 +679,7 @@ def _solve(self):
             # update rho_new if it was updated in _compute_norms
             if rkn is not None:
                 # new rho
-                rhos[-1] = rkn ** 2
+                rhos[-1] = rkn**2
 
             self.iter += 1
 

krypy/utils.py

@@ -359,7 +359,7 @@ def __init__(self, x):
             alpha = 1 if gamma == 0 else gamma / xnorm
         else:
             sigma = numpy.linalg.norm(v[1:], 2)
-            xnorm = numpy.sqrt(numpy.abs(gamma) ** 2 + sigma ** 2)
+            xnorm = numpy.sqrt(numpy.abs(gamma) ** 2 + sigma**2)
 
             # is x the multiple of first unit vector?
             if sigma == 0:
@@ -376,7 +376,7 @@ def __init__(self, x):
                     alpha = -gamma / numpy.abs(gamma)
 
         self.xnorm = xnorm
-        self.v = v / numpy.sqrt(numpy.abs(v[0]) ** 2 + sigma ** 2)
+        self.v = v / numpy.sqrt(numpy.abs(v[0]) ** 2 + sigma**2)
         self.alpha = alpha
         self.beta = beta
 
@@ -771,7 +771,7 @@ def angles(F, G, ip_B=None, compute_vectors=False):
     else:
         Y, s, Z = scipy.linalg.svd(inner(QF, QG, ip_B=ip_B))
         Vcos = numpy.dot(QG, Z.T.conj())
-        n_large = numpy.flatnonzero((s ** 2) < 0.5).shape[0]
+        n_large = numpy.flatnonzero((s**2) < 0.5).shape[0]
         n_small = s.shape[0] - n_large
         theta = numpy.hstack(
             [
@@ -1913,7 +1913,7 @@ def __init__(self, evals, exclude_zeros=False):
 
     def eval_step(self, step):
         """Evaluate bound for given step."""
-        return 2 * self.base ** step
+        return 2 * self.base**step
 
     def get_step(self, tol):
         """Return step at which bound falls below tolerance."""
@@ -2003,7 +2003,7 @@ def eval_step(self, step):
         return 2 * self.base ** numpy.floor(step / 2.0)
 
     def get_step(self, tol):
-        """Return step at which bound falls below tolerance. """
+        """Return step at which bound falls below tolerance."""
         return 2 * numpy.log(tol / 2.0) / numpy.log(self.base)
 
 

test/test_utils.py

@@ -504,7 +504,7 @@ def assert_arnoldi(
     arnoldi_res = MAV - numpy.dot(V, H)
     arnoldi_resn = krypy.utils.norm(arnoldi_res, ip_B=ip_B)
     # inequality (2.3) in [1]
-    arnoldi_tol = arnoldi_const * k * (N ** 1.5) * eps * An
+    arnoldi_tol = arnoldi_const * k * (N**1.5) * eps * An
     assert arnoldi_resn <= arnoldi_tol
 
     # check orthogonality by measuring \| I - <V,V> \|_2
@@ -514,7 +514,7 @@ def assert_arnoldi(
         ortho_res = numpy.eye(V.shape[1]) - krypy.utils.inner(V, V, ip_B=ip_B)
     ortho_resn = numpy.linalg.norm(ortho_res, 2)
     if ortho == "house":
-        ortho_tol = ortho_const * (k ** 1.5) * N * eps  # inequality (2.4) in [1]
+        ortho_tol = ortho_const * (k**1.5) * N * eps  # inequality (2.4) in [1]
     else:
         vAV_singvals = scipy.linalg.svd(
             numpy.column_stack([V[:, [0]], (MAV[:, :-1] if invariant else MAV)]),
@@ -525,7 +525,7 @@ def assert_arnoldi(
         else:
             # inequality (2.5) in [1]
             ortho_tol = (
-                ortho_const * (k ** 2) * N * eps * vAV_singvals[0] / vAV_singvals[-1]
+                ortho_const * (k**2) * N * eps * vAV_singvals[0] / vAV_singvals[-1]
             )
     # mgs or lanczos is not able to detect an invariant subspace reliably
     if (ortho != "mgs" or N != k) and ortho != "lanczos":