Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_init(self):
data = list(range(10))
ts = [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
table = make_electrode_table()
region = DynamicTableRegion('electrodes', [0, 2], 'the first and third electrodes', table)
eS = ElectricalSeries('test_eS', data, region, channel_conversion=[2., 6.3], timestamps=ts)
self.assertEqual(eS.name, 'test_eS')
self.assertEqual(eS.data, data)
self.assertEqual(eS.timestamps, ts)
def test_init(self):
table = make_electrode_table()
region = DynamicTableRegion('electrodes', [0, 2], 'the first and third electrodes', table)
sES = SpikeEventSeries('test_sES', list(range(10)), list(range(10)), region)
ew = EventWaveform(sES)
self.assertEqual(ew.spike_event_series['test_sES'], sES)
self.assertEqual(ew['test_sES'], ew.spike_event_series['test_sES'])
def setUpContainer(self):
data = list(range(10))
ts = [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]
TestElectricalSeriesIO.make_electrode_table(self)
region = DynamicTableRegion('electrodes', [0, 2], 'the first and third electrodes', self.table)
self.eS = ElectricalSeries('test_eS', data, region, timestamps=ts)
eD = EventDetection('detection_method', self.eS, (1, 2, 3), (0.1, 0.2, 0.3))
return eD
def setUpElectricalSeriesContainers(self):
TestElectricalSeriesIO.make_electrode_table(self)
region1 = DynamicTableRegion('electrodes', [0, 2], 'the first and third electrodes', self.table)
region2 = DynamicTableRegion('electrodes', [1, 3], 'the second and fourth electrodes', self.table)
data1 = list(zip(range(10), range(10, 20)))
data2 = list(zip(reversed(range(10)), reversed(range(10, 20))))
timestamps = list(map(lambda x: x/10, range(10)))
es1 = ElectricalSeries('test_eS1', data1, region1, timestamps=timestamps)
es2 = ElectricalSeries('test_eS2', data2, region2, channel_conversion=[4., .4], timestamps=timestamps)
return es1, es2
def setUpTwoElectricalSeries(self):
""" Return two test ElectricalSeries to read/write """
TestElectricalSeriesIO.make_electrode_table(self)
region1 = DynamicTableRegion(name='electrodes',
data=[0, 2],
description='the first and third electrodes',
table=self.table)
region2 = DynamicTableRegion(name='electrodes',
data=[1, 3],
description='the second and fourth electrodes',
table=self.table)
data1 = list(zip(range(10), range(10, 20)))
data2 = list(zip(reversed(range(10)), reversed(range(10, 20))))
timestamps = list(map(lambda x: x/10., range(10)))
es1 = ElectricalSeries(name='test_eS1', data=data1, electrodes=region1, timestamps=timestamps)
es2 = ElectricalSeries(name='test_eS2', data=data2, electrodes=region2, channel_conversion=[4., .4],
timestamps=timestamps)
return es1, es2
def setUpElectricalSeriesContainers(self):
TestElectricalSeriesIO.make_electrode_table(self)
region1 = DynamicTableRegion('electrodes', [0, 2], 'the first and third electrodes', self.table)
region2 = DynamicTableRegion('electrodes', [1, 3], 'the second and fourth electrodes', self.table)
data1 = list(zip(range(10), range(10, 20)))
data2 = list(zip(reversed(range(10)), reversed(range(10, 20))))
timestamps = list(map(lambda x: x/10, range(10)))
es1 = ElectricalSeries('test_eS1', data1, region1, timestamps=timestamps)
es2 = ElectricalSeries('test_eS2', data2, region2, channel_conversion=[4., .4], timestamps=timestamps)
return es1, es2
def setUpContainer(self):
TestElectricalSeriesIO.make_electrode_table(self)
region = DynamicTableRegion('electrodes', [0, 2], 'the first and third electrodes', self.table)
sES = SpikeEventSeries(
'test_sES', ((1, 1, 1), (2, 2, 2)), list(range(2)), region)
ew = EventWaveform(sES)
return ew
def test_add_electrical_series(self):
lfp = LFP()
table = make_electrode_table()
region = DynamicTableRegion('electrodes', [0, 2], 'the first and third electrodes', table)
eS = ElectricalSeries('test_eS', [0, 1, 2, 3], region, timestamps=[0.1, 0.2, 0.3, 0.4])
lfp.add_electrical_series(eS)
self.assertEqual(lfp.electrical_series.get('test_eS'), eS)
def test_invalid_init_mismatched_description(self):
event_times = [1]
table = make_electrode_table()
electrodes = DynamicTableRegion('electrodes', [0, 2], 'the first and third electrodes', table)
description = ['desc1', 'desc2', 'desc3', 'desc4'] # Need 3 descriptions but give 4
features = [[[0, 1, 2], [3, 4, 5]]]
self.assertRaises(ValueError, FeatureExtraction, electrodes, description, event_times, features)
def test_invalid_init_mismatched_event_times(self):
event_times = [] # Need 1 event time but give 0
table = make_electrode_table()
electrodes = DynamicTableRegion('electrodes', [0, 2], 'the first and third electrodes', table)
description = ['desc1', 'desc2', 'desc3']
features = [[[0, 1, 2], [3, 4, 5]]]
self.assertRaises(ValueError, FeatureExtraction, electrodes, description, event_times, features)