Changeset 451 for trunk/test/test_analysis.py

Timestamp:

08/23/10 13:47:57 (21 months ago)

Author:

mpereira

Message:

Added make_kernel() to analysis.py

Files:

• trunk/test/test_analysis.py (modified) (3 diffs)

trunk/test/test_analysis.py

@@ -2 +2 @@
 Unit tests for the NeuroTools.analysis module
 """
+import numpy
+import scipy.io
+import unittest
+from numpy import pi, sin
 
 from NeuroTools import analysis
-import numpy, unittest
-from numpy import pi, sin
+
 
 # test simple_frequency_spectrum
@@ -11 +14 @@
 
 class AnalysisTest(unittest.TestCase):
+
+    def setUp(self):
+        #The following are used for the make_kernel testcases
+        self.box = scipy.io.loadmat('analysis/make_kernel/box.mat')
+        self.tri = scipy.io.loadmat('analysis/make_kernel/tri.mat')
+        self.epa = scipy.io.loadmat('analysis/make_kernel/epa.mat')
+        self.gau = scipy.io.loadmat('analysis/make_kernel/gau.mat')
+        self.alp = scipy.io.loadmat('analysis/make_kernel/alp.mat')
+        self.exp = scipy.io.loadmat('analysis/make_kernel/exp.mat')
+        self.alp_reversed = scipy.io.loadmat(
+            'analysis/make_kernel/alp_reversed.mat')
+        self.exp_reversed = scipy.io.loadmat(
+            'analysis/make_kernel/exp_reversed.mat')
 
     def testSimpleFrequencySpectrum(self):
@@ -36 +52 @@
         z=analysis.ccf(a,a)
         assert z[len(z)/2] == 1
+    
+    def testMakeKernelBox(self):
+        true_kernel = self.box['kernel'].ravel()
+        true_norm = self.box['norm'].ravel()[0]
+        true_m_idx = self.box['m_idx'].ravel()[0] - 1
+        
+        kernel, norm, m_idx = analysis.make_kernel('BOX', 1, 0.001)
+        
+        numpy.testing.assert_array_almost_equal(true_kernel, kernel, decimal = 3
+                                                )
+        self.assertEqual(true_norm, norm, 3)
+        self.assertEqual(true_m_idx, m_idx, 3)
+        
+    def testMakeKernelTri(self):
+        true_kernel = self.tri['kernel'].ravel()
+        true_norm = self.tri['norm'].ravel()[0]
+        true_m_idx = self.tri['m_idx'].ravel()[0] - 1
+        
+        kernel, norm, m_idx = analysis.make_kernel('TRI', 1, 0.001)
+        
+        numpy.testing.assert_array_almost_equal(true_kernel, kernel, decimal = 3
+                                                )
+        self.assertEqual(true_norm, norm, 3)
+        self.assertEqual(true_m_idx, m_idx, 3)
+        
+    def testMakeKernelEpa(self):
+        true_kernel = self.epa['kernel'].ravel()
+        true_norm = self.epa['norm'].ravel()[0]
+        true_m_idx = self.epa['m_idx'].ravel()[0] - 1
+        
+        kernel, norm, m_idx = analysis.make_kernel('EPA', 1, 0.001)
+        
+        numpy.testing.assert_array_almost_equal(true_kernel, kernel, decimal = 3
+                                                )
+        self.assertEqual(true_norm, norm, 3)
+        self.assertEqual(true_m_idx, m_idx, 3)
+        
+    def testMakeKernelGau(self):
+        true_kernel = self.gau['kernel'].ravel()
+        true_norm = self.gau['norm'].ravel()[0]
+        true_m_idx = self.gau['m_idx'].ravel()[0] - 1
+        
+        kernel, norm, m_idx = analysis.make_kernel('GAU', 1, 0.001)
+        
+        numpy.testing.assert_array_almost_equal(true_kernel, kernel, decimal = 3
+                                                )
+        self.assertEqual(true_norm, norm, 3)
+        self.assertEqual(true_m_idx, m_idx, 3)
+        
+    def testMakeKernelAlp(self):
+        true_kernel = self.alp['kernel'].ravel()
+        true_norm = self.alp['norm'].ravel()[0]
+        true_m_idx = self.alp['m_idx'].ravel()[0] - 1
+        
+        kernel, norm, m_idx = analysis.make_kernel('ALP', 1, 0.001)
+        
+        numpy.testing.assert_array_almost_equal(true_kernel, kernel, decimal = 3
+                                                )
+        self.assertEqual(true_norm, norm, 3)
+        self.assertEqual(true_m_idx, m_idx, 3)
 
+    def testMakeKernelExp(self):
+        true_kernel = self.exp['kernel'].ravel()
+        true_norm = self.exp['norm'].ravel()[0]
+        true_m_idx = self.exp['m_idx'].ravel()[0] - 1
+        
+        kernel, norm, m_idx = analysis.make_kernel('EXP', 1, 0.001)
+        
+        numpy.testing.assert_array_almost_equal(true_kernel, kernel, decimal = 3
+                                                )
+        self.assertEqual(true_norm, norm, 3)
+        self.assertEqual(true_m_idx, m_idx, 3)
+    
+    def testMakeKernelAlpReversed(self):
+        """Same as testAlp but with direction = -1
+        """
+        true_kernel = self.alp_reversed['kernel'].ravel()
+        true_norm = self.alp_reversed['norm'].ravel()[0]
+        true_m_idx = self.alp_reversed['m_idx'].ravel()[0] - 1
+        
+        kernel, norm, m_idx = analysis.make_kernel('ALP', 1, 0.001,
+                                                  direction = -1)
+        numpy.testing.assert_array_almost_equal(true_kernel, kernel, decimal = 3
+                                                )
+        self.assertEqual(true_norm, norm, 3)
+        self.assertEqual(true_m_idx, m_idx, 3)
+        
+    def testMakeKernelExpReversed(self):
+        """Same as testExp but with direction = -1
+        """
+        true_kernel = self.exp_reversed['kernel'].ravel()
+        true_norm = self.exp_reversed['norm'].ravel()[0]
+        true_m_idx = self.exp_reversed['m_idx'].ravel()[0] - 1
+        
+        kernel, norm, m_idx = analysis.make_kernel('EXP', 1, 0.001, direction = -1)
+        
+        numpy.testing.assert_array_almost_equal(true_kernel, kernel, decimal = 3
+                                                )
+        self.assertEqual(true_norm, norm, 3)
+        self.assertEqual(true_m_idx, m_idx, 3)
+        
 if __name__ == "__main__":
     unittest.main()