black

LotkaVolterraClass.py

@@ -55,6 +55,7 @@ def LyapunovFunction():
         # TODO: https://www.sciencedirect.com/science/article/pii/S1474667017354101
         raise NotImplementedError()
 
+
 if __name__ == "__main__":
     # Definition of parameters
     a = 1.0  # natural growth rate of rabbits (prey)
@@ -108,5 +109,4 @@ def LyapunovFunction():
 
     fig1.savefig("rabbits_and_foxes" + str(numiter) + ".png")
 
-
     plt.show()

test_LVC.py

@@ -14,20 +14,18 @@
 numiter = 10000
 dt = 1.0 / numiter
 
+
 def test_sameresult():
     lvm = LVM(a, b, c, d, dt)
     saver = DataSaver(lvm)
 
-
     X0 = np.array([10, 5])  # initial conditions: 10 rabbits and 5 foxes
     lvm.dynamics(X0, numiter, saver)
     T, Xeuler = saver.get_data()["state_T"], np.array(saver.get_data()["state_X"])
     rabbits, foxes = Xeuler.T
-    
+
     X, infodict = integrate.odeint(
-        lambda x, _: dX_dt(x, a, b, c, d), X0,T, full_output=True
+        lambda x, _: dX_dt(x, a, b, c, d), X0, T, full_output=True
     )
 
-    assert np.allclose(Xeuler , X,atol=1e-3)
-
-
+    assert np.allclose(Xeuler, X, atol=1e-3)