[EDF] Add context manager capabilities, __del__ and test

Author: sprenger

neo/rawio/edfrawio.py

@@ -223,5 +223,21 @@ def _rescale_event_timestamp(self, event_timestamps, dtype, event_channel_index)
     def _rescale_epoch_duration(self, raw_duration, dtype, event_channel_index):
         return None
 
+    def __enter__(self):
+        return self
+
+    def __del__(self):
+        self._close_reader()
+
+    def __exit__(self, exc_type, exc_val, ex_tb):
+        self._close_reader()
+
     def close(self):
-        self.edf_reader.close()
+        """
+        Closes the file handler
+        """
+        self._close_reader()
+
+    def _close_reader(self):
+        if hasattr(self, 'edf_reader'):
+            self.edf_reader.close()

neo/test/rawiotest/test_edfrawio.py

@@ -15,6 +15,29 @@ class TestExampleRawIO(BaseTestRawIO, unittest.TestCase, ):
         'edf/edf+C.edf',
     ]
 
+    def test_context_handler(self):
+        filename = self.get_local_path('edf/edf+C.edf')
+        with EDFRawIO(filename) as io:
+            io.parse_header()
+
+        # Check that file was closed properly and can be opened again
+        with open(filename) as f:
+            pass
+
+
+    def test_close(self):
+        filename = self.get_local_path('edf/edf+C.edf')
+
+        # Open file and close it again
+        io1 = EDFRawIO(filename)
+        io1.parse_header()
+        io1.close()
+
+        # Check that file was closed properly and can be opened again
+        io2 = EDFRawIO(filename)
+        io2.parse_header()
+        io2.close()
+
 
 if __name__ == "__main__":
     unittest.main()