From 843b07785058af2f75cb0018d739e17b4f8d14eb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= <dedebenui@protonmail.com>
Date: Thu, 29 Feb 2024 14:08:04 +0100
Subject: [PATCH] misc, tests

---
 src/scgenerator/__init__.py  |  2 +-
 src/scgenerator/threading.py |  1 +
 tests/test_cache.py          | 11 +++++++++--
 tests/test_spectra.py        | 12 ++++++++++++
 4 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/src/scgenerator/__init__.py b/src/scgenerator/__init__.py
index b9bd287..0249993 100644
--- a/src/scgenerator/__init__.py
+++ b/src/scgenerator/__init__.py
@@ -10,4 +10,4 @@ from scgenerator.spectra import Spectrum, propagation, propagation_series
 from scgenerator.physics import fiber, materials, plasma, pulse
 from scgenerator.physics.units import PlotRange
 from scgenerator.solver import SimulationResult, integrate, solve43
-from scgenerator.cache import Cache
+from scgenerator.cache import Cache, normalize_path
diff --git a/src/scgenerator/threading.py b/src/scgenerator/threading.py
index 830e198..6412298 100644
--- a/src/scgenerator/threading.py
+++ b/src/scgenerator/threading.py
@@ -141,6 +141,7 @@ class MultibarThread(Thread):
             position=0,
             colour="green",
             desc=desc,
+            smoothing=0,
         )
         self.started = False
         self.finished = False
diff --git a/tests/test_cache.py b/tests/test_cache.py
index 90ebf56..5cc29dc 100644
--- a/tests/test_cache.py
+++ b/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"
+    )
diff --git a/tests/test_spectra.py b/tests/test_spectra.py
index e182816..ac838e1 100644
--- a/tests/test_spectra.py
+++ b/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