image data helper in plotting

2024-01-08 11:30:22 +01:00
parent 2d1cf0c9c1
commit eafb88a899
2 changed files with 26 additions and 20 deletions
--- a/tests/test_integrator.py
+++ b/tests/test_integrator.py
@@ -60,7 +60,7 @@ def test_rk43_soliton(plot=False):

    res = sc.integrate(spec0, end, lin, non_lin, targets=targets, atol=1e-10, rtol=1e-9)
    if plot:
-        x, y, z = sc.plotting.transform_2D_propagation(
+        x, y, z = sc.plotting.transform_2D_data(
            res.spectra,
            sc.PlotRange(500, 1300, "nm"),
            w_c + w0,