Merge pull request #39 from AUAS-Pulsar/dedisp-tests

add test for dedisp

Author: TVH7

dedisperse/dedisperse.py

@@ -8,7 +8,7 @@ def dedisperse(samples, dm):
     '''
     This method performs dedispersion on the filterbank data
     '''
-
+    samples = np.asarray(samples)
     # Distribute the DM over the amount of samples
     delays_per_sample = np.round(np.linspace(dm, 0, samples.shape[1])).astype(int)
 

tests/context.py

@@ -11,5 +11,6 @@
 import plot
 import fourier
 import clipping
+import dedisperse
 import filterbank.header as header
 import filterbank.filterbank as filterbank

tests/test_dedisperse.py

@@ -0,0 +1,29 @@
+"""
+    dedisperse.py unit tests
+"""
+
+import unittest
+
+from .context import dedisperse # pylint: disable-msg=E0611
+
+class TestDedisperse(unittest.TestCase):
+    """
+        Class for testing dedisperse.py
+    """
+
+    samples = [[10, 1, 1, 1, 1, 1, 1],
+               [1, 10, 1, 1, 1, 1, 1],
+               [1, 1, 10, 1, 1, 1, 1],
+               [1, 1, 1, 10, 1, 1, 1],
+               [1, 1, 1, 1, 10, 1, 1],
+               [1, 1, 1, 1, 1, 10, 1],
+               [1, 1, 1, 1, 1, 1, 10]]
+
+    def test_dedisperse(self):
+        """
+            When performining dedispersion,
+            expect moved frequencies per sample
+        """
+        disp_measure = 6
+        results = dedisperse.dedisperse(self.samples, disp_measure)
+        self.assertListEqual(list(results[len(self.samples)-1]), [10]*7)