add tests for bayesian blocks

mutis/flares/bayblocks.py

@@ -213,7 +213,7 @@ def get_flare_list(self):
                           np.where((np.abs(np.diff(np.asarray(self.inflare, dtype=int))) == 1))[0]+1)
 
         # (groups contains the groups of indices of self.inflare/self.values 
-        # that corresond to flares or no flares)
+        # that correspond to flares or no flares)
 
         #print(self.inflare)
         #print([list(grp) for grp in groups])

mutis/tests/test_bayblocks.py

@@ -0,0 +1,16 @@
+import unittest
+import numpy as np
+from mutis.flares.bayblocks import BayesianBlocks
+from mutis import Signal
+
+def test_bayblocks():
+    # test the bayblock algorithm with a signal that corresponds to 3 gaussian pulses in a row
+    t = np.linspace(0, 6, 50)
+    y = np.exp(-(t-1)**2/0.1) + np.exp(-(t-3)**2/0.1) + np.exp(-(t-5)**2/0.1)
+    dy = 0.1 * np.abs(y) 
+    signal = Signal(t, y, dy)
+    bayblocks = BayesianBlocks(signal, p=1e-1)
+    bayblocks.get_flares()
+    bayblocks.signal.plot()
+    bayblocks.plot()
+    assert len(bayblocks.get_flare_list()) == 3