def chaining_test():
data = {
'fs': 512,
'num_channels': 1,
'channel_names': ['sine1'],
'epochs': 1
}
sig = SignalManager(generator=sine_wave(data=data))
sig.extract_time_range(0, 2)\
.extract_time_range(0, 1)
def extract_time_range_test(self):
data = {
'fs': 512,
'num_channels': 1,
'channel_names': ['sine1'],
'epochs': 1
}
sig = SignalManager(generator=sine_wave(data=data))
sig.extract_time_range(0, 1)
self.assertEqual(sig.data.shape, (1, 1, 512))