def test_time_equal(self): #import numpy av = AnalogSignalArray(numpy.array([[0,1,2,3,4,5],[0,1,2,3,4,5]]).T,sampling_rate=1.0*pq.Hz, units='mV') t_start = 0 * pq.s t_stop = 6 * pq.s av2 = av.time_slice(t_start,t_stop) ar = numpy.array([a == b for (a,b) in zip(av2.flatten(),av.flatten())]) self.assertEqual(ar.all(),True) self.assertIsInstance(av2, AnalogSignalArray) self.assertEqual(av2.t_stop,t_stop) self.assertEqual(av2.t_start,t_start)
def test_time_slice_offset(self): #import numpy av = AnalogSignalArray(numpy.array([[0,1,2,3,4,5],[0,1,2,3,4,5]]).T, t_start = 10.0 * pq.s,sampling_rate=1.0*pq.Hz, units='mV') t_start = 12 * pq.s t_stop = 14 * pq.s av2 = av.time_slice(t_start,t_stop) correct = AnalogSignalArray(numpy.array([[2,3],[2,3]]).T, t_start = 12.0*pq.ms,sampling_rate=1.0*pq.Hz, units='mV') ar = numpy.array([a == b for (a,b) in zip(av2.flatten(),correct.flatten())]) self.assertEqual(ar.all(),True) self.assertIsInstance(av2, AnalogSignalArray) self.assertEqual(av2.t_stop,t_stop) self.assertEqual(av2.t_start,t_start) self.assertEqual(av2.sampling_rate, correct.sampling_rate)
def test_time_equal(self): #import numpy av = AnalogSignalArray(numpy.array([[0, 1, 2, 3, 4, 5], [0, 1, 2, 3, 4, 5]]).T, sampling_rate=1.0 * pq.Hz, units='mV') t_start = 0 * pq.s t_stop = 6 * pq.s av2 = av.time_slice(t_start, t_stop) ar = numpy.array( [a == b for (a, b) in zip(av2.flatten(), av.flatten())]) self.assertEqual(ar.all(), True) self.assertIsInstance(av2, AnalogSignalArray) self.assertEqual(av2.t_stop, t_stop) self.assertEqual(av2.t_start, t_start)
def test_time_slice_offset(self): #import numpy av = AnalogSignalArray(numpy.array([[0, 1, 2, 3, 4, 5], [0, 1, 2, 3, 4, 5]]).T, t_start=10.0 * pq.s, sampling_rate=1.0 * pq.Hz, units='mV') t_start = 12 * pq.s t_stop = 14 * pq.s av2 = av.time_slice(t_start, t_stop) correct = AnalogSignalArray(numpy.array([[2, 3], [2, 3]]).T, t_start=12.0 * pq.ms, sampling_rate=1.0 * pq.Hz, units='mV') ar = numpy.array( [a == b for (a, b) in zip(av2.flatten(), correct.flatten())]) self.assertEqual(ar.all(), True) self.assertIsInstance(av2, AnalogSignalArray) self.assertEqual(av2.t_stop, t_stop) self.assertEqual(av2.t_start, t_start) self.assertEqual(av2.sampling_rate, correct.sampling_rate)