Changeset 453 for trunk/test/test_analysis.py

Timestamp:

08/23/10 14:25:44 (21 months ago)

Author:

mpereira

Message:

Added crosscorrelate() to analysis.py

Files:

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

trunk/test/test_analysis.py

@@ -7 +7 @@
 from numpy import pi, sin
 
-from NeuroTools import analysis
-
+from NeuroTools import analysis, signals
 
 # test simple_frequency_spectrum
@@ -27 +26 @@
         self.exp_reversed = scipy.io.loadmat(
             'analysis/make_kernel/exp_reversed.mat')
+        
+        #Used for the testcases of the crosscorrelate function
+        spk = signals.load_spikelist('analysis/crosscorrelate/spike_data')
+        self.spk0 = spk[0].spike_times
+        self.spk1 = spk[1].spike_times
 
     def testSimpleFrequencySpectrum(self):
@@ -153 +157 @@
         self.assertEqual(true_m_idx, m_idx, 3)
         
+    def testCrosscorrelateNoLag(self):
+            int, int_, norm = analysis.crosscorrelate(self.spk0, self.spk1)
+            #The following are output was generated with the FIND MATLAB toolbox
+            matlab_int = numpy.loadtxt('analysis/crosscorrelate/out_matlab_int')
+            numpy.testing.assert_array_almost_equal(int, matlab_int,
+                                                    decimal=3)
+            #The int_ output has a random component and for this reason the test
+            # cases are not as trivial
+            
+    def testCrosscorrelateLag100(self):
+        """Test case with lag within the length of the input array
+        """
+        int, int_, norm, = analysis.crosscorrelate(self.spk0, self.spk1,
+                                                   lag=100.0)
+        matlab_int = numpy.loadtxt('analysis/crosscorrelate/out_matlab_int_lag_100')
+        numpy.testing.assert_array_almost_equal(int, matlab_int, decimal = 3)
+            
+    def testCrosscorrelateLag500(self):
+        """Test case with lag is higher than the trial length
+        """
+        int, int_, norm = analysis.crosscorrelate(self.spk0, self.spk1,
+                                                  lag=500.0)
+        matlab_int = numpy.loadtxt('analysis/crosscorrelate/out_matlab_int_lag_500')
+        numpy.testing.assert_array_almost_equal(int, matlab_int, decimal = 3)
+            
 if __name__ == "__main__":
     unittest.main()