Small improvements

tests/Optica_PM2000D/test_Optica_PM2000D.py

@@ -1,3 +1,9 @@
+"""
+May 2023
+
+Testing the new solver / operators structure
+using parameters from the 2022 Optica paper
+"""
 import matplotlib.pyplot as plt
 import numpy as np
 from scipy.interpolate import interp1d
@@ -19,11 +25,12 @@ def main():
     #
     # plt.plot(params.w, params.linear_operator(0).imag)
     # plt.show()
-    
 
     breakpoint()
 
-    res = sol.integrate(params.spec_0, params.length, params.linear_operator, params.nonlinear_operator)
+    res = sol.integrate(
+        params.spec_0, params.length, params.linear_operator, params.nonlinear_operator
+    )
 
     new_z = np.linspace(0, params.length, 256)