black tests passed

Author: nmank

pydmd/dmdc.py

@@ -191,12 +191,14 @@ class DMDc(DMDBase):
         equal than `svd_rank`. For the possible values please refer to the
         `svd_rank` parameter description above.
     :type svd_rank_omega: int or float
-    :param lag: the time lag for the snapshots. Used in fit method to generate 
+    :param lag: the time lag for the snapshots. Used in fit method to generate
         X and Y.
     :type lag: int
     """
 
-    def __init__(self, svd_rank=0, tlsq_rank=0, opt=False, svd_rank_omega=-1, lag=1):
+    def __init__(
+        self, svd_rank=0, tlsq_rank=0, opt=False, svd_rank_omega=-1, lag=1
+    ):
         # we're going to initialize Atilde when we know if B is known
         self._Atilde = None
         # remember the arguments for when we'll need them
@@ -309,12 +311,10 @@ def fit(self, X, I, B=None):
         n_samples = self.snapshots.shape[-1]
 
         if self._lag < 1:
-            raise ValueError(
-            "Time lag must be positive."
-                )
-        
-        X = self.snapshots[:, :-self._lag]
-        Y = self.snapshots[:, self._lag:]
+            raise ValueError("Time lag must be positive.")
+
+        X = self.snapshots[:, : -self._lag]
+        Y = self.snapshots[:, self._lag :]
 
         self._set_initial_time_dictionary(
             {"t0": 0, "tend": n_samples - 1, "dt": 1}

tests/test_dmdc.py

@@ -34,8 +34,8 @@ def create_system_without_B_lag(lag=1):
     snapshots = []
     for _ in range(lag):
         snapshots.append(np.array([0.25] * n))
-    u = np.random.rand(n, m-lag) - 0.5
-    for i in range(m-lag):
+    u = np.random.rand(n, m - lag) - 0.5
+    for i in range(m - lag):
         snapshots.append(A.dot(snapshots[i]) + B.dot(u[:, i]))
     snapshots = np.array(snapshots).T
     return {"snapshots": snapshots, "u": u, "B": B, "A": A}
@@ -106,7 +106,7 @@ def test_reconstruct_b_unknown():
 
 def test_reconstruct_b_unknown_lag():
     lag = 3
-    system = create_system_without_B_lag(lag = lag)
+    system = create_system_without_B_lag(lag=lag)
     dmdc = DMDc(svd_rank=-1, opt=True, lag=lag)
     dmdc.fit(system["snapshots"], system["u"])
     np.testing.assert_array_almost_equal(
@@ -276,7 +276,7 @@ def test_correct_amplitudes():
 
 
 def test_lag_param_b_unknown_raises():
-    system = create_system_without_B_lag(lag = 3)
+    system = create_system_without_B_lag(lag=3)
     dmdc = DMDc(svd_rank=-1, opt=True, lag=0)
     with raises(ValueError):
-        dmdc.fit(system["snapshots"], system["u"])
+        dmdc.fit(system["snapshots"], system["u"])