Skip to content

Commit 3265603

Browse files
author
sprenger
committed
[edf] remove io initialization from setUp to avoid parallel ios
1 parent fdf700f commit 3265603

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

neo/test/iotest/test_edfio.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -21,14 +21,14 @@ class TestEDFIO(BaseTestIO, unittest.TestCase, ):
2121

2222
def setUp(self):
2323
super().setUp()
24-
filename = self.get_local_path('edf/edf+C.edf')
25-
self.io = EDFIO(filename)
24+
self.filename = self.get_local_path('edf/edf+C.edf')
2625

2726
def test_read_block(self):
2827
"""
2928
Test reading the complete block and general annotations
3029
"""
31-
bl = self.io.read_block()
30+
io = EDFIO(self.filename)
31+
bl = io.read_block()
3232
self.assertTrue(bl.annotations)
3333

3434
seg = bl.segments[0]
@@ -40,7 +40,8 @@ def test_read_segment_with_time_slice(self):
4040
"""
4141
Test loading of a time slice and check resulting times
4242
"""
43-
seg = self.io.read_segment(time_slice=None)
43+
io = EDFIO(self.filename)
44+
seg = io.read_segment(time_slice=None)
4445

4546
self.assertEqual(len(seg.spiketrains), 0)
4647
self.assertEqual(len(seg.events), 0)
@@ -50,7 +51,7 @@ def test_read_segment_with_time_slice(self):
5051
self.assertEqual(n_channels, 5)
5152

5253
t_start, t_stop = 500 * pq.ms, 800 * pq.ms
53-
seg = self.io.read_segment(time_slice=(t_start, t_stop))
54+
seg = io.read_segment(time_slice=(t_start, t_stop))
5455

5556
self.assertAlmostEqual(seg.t_start.rescale(t_start.units), t_start, delta=5.)
5657
self.assertAlmostEqual(seg.t_stop.rescale(t_stop.units), t_stop, delta=5.)
@@ -59,17 +60,17 @@ def test_compare_data(self):
5960
"""
6061
Compare data from AnalogSignal with plain data stored in text file
6162
"""
62-
63-
plain_data = np.loadtxt(self.io.filename.replace('.edf', '.txt'))
64-
seg = self.io.read_segment(signal_group_mode='split-all')
63+
io = EDFIO(self.filename)
64+
plain_data = np.loadtxt(io.filename.replace('.edf', '.txt'))
65+
seg = io.read_segment(signal_group_mode='split-all')
6566

6667
anasigs = seg.analogsignals
6768

6869
for aidx, anasig in enumerate(anasigs):
6970
ana_data = anasig.magnitude.flatten()
7071

7172
# reverting gain and offset conversion
72-
sig_dict = self.io.signal_headers[aidx]
73+
sig_dict = io.signal_headers[aidx]
7374
physical_range = sig_dict['physical_max'] - sig_dict['physical_min']
7475
digital_range = sig_dict['digital_max'] - sig_dict['digital_min']
7576
gain = physical_range / digital_range

0 commit comments

Comments
 (0)