def test_remove_mean_single_channel(self): tb = generate_timebase(t0=-0.5, n_samples=1.e2, sample_freq=1.e2) # nonzero signal mean tsd = TimeseriesData(timebase=tb, signal=Signal(np.arange(len(tb))), channels=ChannelList(Channel('ch_01',Coords('dummy',(0,0,0))))) filtered_tsd = tsd.subtract_mean() assert_almost_equal(np.mean(filtered_tsd.signal), 0)
def testFilteredDataHistory_nocopy(self): tb = generate_timebase(t0=-0.5, n_samples=1.e2, sample_freq=1.e2) # nonzero signal mean ch = get_n_channels(1) tsd = TimeseriesData(timebase=tb, signal=Signal(np.arange(len(tb))), channels=ch) filtered_tsd = tsd.subtract_mean() #self.assertEqual(len(filtered_tsd.history.split('\n')), 3) self.assertEqual(len(filtered_tsd.history.split('\n')), 5) # bdb thinks extra 2 is OK output_data = filtered_tsd.normalise(method='rms', copy=False) #self.assertEqual(filtered_tsd.history.split('> ')[-1], "normalise(method='rms')") self.assertEqual(filtered_tsd.history.split('> ')[-1].split('\n')[0], "normalise(method='rms')") #self.assertEqual(output_data.history.split('> ')[-1], "normalise(method='rms')") self.assertEqual(output_data.history.split('> ')[-1].split('\n')[0], "normalise(method='rms')")
def testFilteredDataHistory_copy(self): """ make sure that _copy version does NOT alter original """ tb = generate_timebase(t0=-0.5, n_samples=1.e2, sample_freq=1.e2) # nonzero signal mean ch = get_n_channels(1) tsd = TimeseriesData(timebase=tb, signal=Signal(np.arange(len(tb))), channels=ch) filtered_tsd = tsd.subtract_mean() # bdb in 4 places, assume that the xtra info (norm_value) is supposed to be there. #self.assertEqual(len(filtered_tsd.history.split('\n')), 3) self.assertEqual(len(filtered_tsd.history.split('\n')), 5) # bdb thinks extra 2 is OK output_data = filtered_tsd.normalise(method='rms', copy=True) #self.assertEqual(output_data.history.split('> ')[-1], "normalise(method='rms')") self.assertEqual(output_data.history.split('> ')[-1].split('\n')[0], "normalise(method='rms')") #self.assertEqual(filtered_tsd.history.split('> ')[-1], "subtract_mean()") self.assertEqual(filtered_tsd.history.split('> ')[-1].split('\n')[0], "subtract_mean()")