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)