Add test for Data.burst_sizes()

OpenSMFS · Jan 10, 2015 · 45bda43 · 45bda43
1 parent 2e525c2
commit 45bda43
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/fretbursts/tests/test_burstlib.py b/fretbursts/tests/test_burstlib.py
@@ -244,6 +244,21 @@ def test_burst_search(data):
 
     data.burst_search(L=10, m=10, F=7)
 
+def test_burst_sizes(data):
+    """Test for .burst_sizes_ich() and burst_sizes()"""
+    # Smoke test
+    plain_sizes = data.burst_sizes()
+    assert len(plain_sizes) == data.nch
+    # Test gamma and gamma1 arguments
+    bs1 = data.burst_sizes_ich(gamma=0.5)
+    bs2 = data.burst_sizes_ich(gamma1=0.5)
+    assert np.allclose(bs1, bs2/0.5)
+    # Test add_naa
+    if data.ALEX:
+        bs_no_naa = data.burst_sizes_ich(add_naa=False)
+        bs_naa = data.burst_sizes_ich(add_naa=True)
+        assert np.allclose(bs_no_naa + data.naa_, bs_naa)
+
 def test_leakage(data):
     """
     Test setting leakage before and after burst search