add unit tests for threshold and correlate

pycbc/filter/matchedfilter.py

@@ -454,5 +454,5 @@ def overlap_cplx(vec1, vec2, psd=None, low_frequency_cutoff=None,
 
 __all__ = ['match', 'matched_filter', 'sigmasq', 'sigma',
            'sigmasq_series', 'make_frequency_series', 'overlap', 'overlap_cplx',
-           'matched_filter_core']
+           'matched_filter_core', 'correlate']
 

test/test_correlate.py

@@ -0,0 +1,67 @@
+# Copyright (C) 2012  Alex Nitz, Josh Willis
+#
+# This program is free software; you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by the
+# Free Software Foundation; either version 3 of the License, or (at your
+# option) any later version.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
+# Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+
+#
+# =============================================================================
+#
+#                                   Preamble
+#
+# =============================================================================
+#
+"""
+These are the unittests for the pycbc.waveform module
+"""
+import sys
+import pycbc
+import unittest
+import numpy
+from pycbc.types import *
+from pycbc.scheme import *
+from pycbc.filter import *
+import pycbc.fft
+from utils import parse_args_all_schemes, simple_exit
+
+_scheme, _context = parse_args_all_schemes("correlate")
+
+from pycbc.filter.matchedfilter_cpu import correlate_numpy
+trusted_correlate = correlate_numpy
+
+class Testcorrelate(unittest.TestCase):
+    def setUp(self,*args):
+        self.context = _context
+        self.scheme = _scheme
+        self.tolerance = 1e-7
+        xr = numpy.random.uniform(low=-1, high=1.0, size=2**20)
+        yr = numpy.random.uniform(low=-1, high=1.0, size=2**20)
+        xi = numpy.random.uniform(low=-1, high=1.0, size=2**20)
+        yi = numpy.random.uniform(low=-1, high=1.0, size=2**20)
+        self.x = Array(xr + xi * 1.0j, dtype=complex64)
+        self.y = Array(yr + yi * 1.0j, dtype=complex64)
+        self.z = zeros(2**20, dtype=complex64)
+        trusted_correlate(self.x, self.y, self.z)
+
+    def test_correlate(self):
+        with self.context:
+            z = zeros(2**20, dtype=complex64)
+            correlate(self.x, self.y, z)
+            self.assertTrue(self.z.almost_equal_elem(z, self.tolerance))
+
+suite = unittest.TestSuite()
+suite.addTest(unittest.TestLoader().loadTestsFromTestCase(Testcorrelate))
+
+if __name__ == '__main__':
+    results = unittest.TextTestRunner(verbosity=2).run(suite)
+    simple_exit(results)

test/test_threshold.py

@@ -0,0 +1,66 @@
+# Copyright (C) 2012  Alex Nitz, Josh Willis
+#
+# This program is free software; you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by the
+# Free Software Foundation; either version 3 of the License, or (at your
+# option) any later version.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General
+# Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+
+#
+# =============================================================================
+#
+#                                   Preamble
+#
+# =============================================================================
+#
+"""
+These are the unittests for the pycbc.waveform module
+"""
+import sys
+import pycbc
+import unittest
+import numpy
+from pycbc.types import *
+from pycbc.scheme import *
+from pycbc.events import *
+import pycbc.fft
+from utils import parse_args_all_schemes, simple_exit
+
+_scheme, _context = parse_args_all_schemes("Threshold")
+
+from pycbc.events.threshold_cpu import threshold_numpy
+trusted_threshold = threshold_numpy
+
+class TestThreshold(unittest.TestCase):
+    def setUp(self,*args):
+        self.context = _context
+        self.scheme = _scheme
+        r = numpy.random.uniform(low=-1, high=1.0, size=2**20)
+        i = numpy.random.uniform(low=-1, high=1.0, size=2**20)
+        v = r + i*1.0j
+        self.series = Array(v, dtype=complex64)
+        self.threshold = 1.3
+        self.locs, self.vals = trusted_threshold(self.series, self.threshold) 
+        self.tolerance = 1e-6
+        print len(self.locs), len(self.vals)
+
+    def test_threshold(self):
+        with self.context:
+            locs, vals = threshold(self.series, self.threshold)
+            self.assertTrue((locs == self.locs).all())
+            self.assertTrue((vals == self.vals).all())
+            print len(locs), len(vals)
+suite = unittest.TestSuite()
+suite.addTest(unittest.TestLoader().loadTestsFromTestCase(TestThreshold))
+
+if __name__ == '__main__':
+    results = unittest.TextTestRunner(verbosity=2).run(suite)
+    simple_exit(results)