@@ -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