misc, tests

tests/test_cache.py

@@ -1,5 +1,5 @@
 import scgenerator.cache as scc
-from scgenerator.cache import Cache, methodcache, normalize_path
+from scgenerator.cache import Cache, normalize_path
 
 
 def test_io():
@@ -74,7 +74,6 @@ class Stuff:
 
 def test_decorator():
     cache = Cache("Test")
-    cache.compile_class(Stuff)
     cache.delete()
     call_count = 0
 
@@ -125,3 +124,11 @@ def test_normalize():
         )
         == "sdfl asdfk asd"
     )
+    assert (
+        normalize_path(
+            "i=-1 [8.50e-07 9.00e-07 9.44e-07 1.01e-06 1.00e-06 1.05e-06 "
+            "1.10e-06 1.15e-06\n 1.20e-06 1.25e-06] [1.5e-08]nm.scbin"
+        )
+        == "i=-1 [8.50e-07 9.00e-07 9.44e-07 1.01e-06 1.00e-06 1.05e-06 1.10e-06 1.15e-06 "
+        "1.20e-06 1.25e-06] [1.5e-08]nm.scbin"
+    )

tests/test_spectra.py

@@ -85,3 +85,15 @@ def test_energy():
     spec = Spectrum(spec, w, t)
     assert np.all(spec.energy == np.trapz(abs2(np.fft.ifft(spec)), x=t, axis=-1))
     assert spec.energy.shape == (3, 2)
+
+
+def test_mask():
+    wl = 800e-9
+    bw = 50e-9
+    t = np.linspace(-1e-12, 1e-12, 4096)
+    w = wspace(t) + m_rads(wl)
+    spec = Spectrum(np.ones(4096) + 0j, w, t)
+    mased = spec.mask_wl(wl, bw)
+    above = mased.wl_int / mased.wl_int.max() - 0.5 >= 0
+    diff = spec.wl_disp[above].max() - spec.wl_disp[above].min()
+    assert diff == bw