update neuroshare tests to also work for neuroshareapiio

Author: JuliaSprenger

neo/test/iotest/test_neuroshareio.py

@@ -4,6 +4,8 @@
 
 import sys
 import os
+import pathlib
+import shutil
 import tarfile
 import zipfile
 import tempfile
@@ -15,34 +17,33 @@
 from neo.test.iotest.common_io_test import BaseTestIO
 
 
-class TestNeuroshareIO(unittest.TestCase, BaseTestIO):
+class TestNeuroshareIO(BaseTestIO, unittest.TestCase):
     ioclass = NeuroshareIO
-    entities_to_download = [
-        'neuroshare/Multichannel_fil_1.mcd'
-    ]
+    entities_to_download = ['neuroshare']
     entities_to_test = ['neuroshare/Multichannel_fil_1.mcd']
 
-    def setUp(self):
-        BaseTestIO.setUp(self)
+    @classmethod
+    def setUpClass(cls, *args, **kwargs):
+        super().setUpClass()
         if sys.platform.startswith('win'):
             distantfile = 'http://download.multichannelsystems.com/download_data/software/neuroshare/nsMCDLibrary_3.7b.zip'
             localfile = os.path.join(tempfile.gettempdir(), 'nsMCDLibrary_3.7b.zip')
             if not os.path.exists(localfile):
                 urlretrieve(distantfile, localfile)
             if platform.architecture()[0].startswith('64'):
-                self.dllname = os.path.join(
+                cls.dllname = os.path.join(
                     tempfile.gettempdir(),
                     'Matlab/Matlab-Import-Filter/Matlab_Interface/nsMCDLibrary64.dll')
-                if not os.path.exists(self.dllname):
+                if not os.path.exists(cls.dllname):
                     zip = zipfile.ZipFile(localfile)
                     zip.extract(
                         'Matlab/Matlab-Import-Filter/Matlab_Interface/nsMCDLibrary64.dll',
                         path=tempfile.gettempdir())
             else:
-                self.dllname = os.path.join(
+                cls.dllname = os.path.join(
                     tempfile.gettempdir(),
                     'Matlab/Matlab-Import-Filter/Matlab_Interface/nsMCDLibrary.dll')
-                if not os.path.exists(self.dllname):
+                if not os.path.exists(cls.dllname):
                     zip = zipfile.ZipFile(localfile)
                     zip.extract(
                         'Matlab/Matlab-Import-Filter/Matlab_Interface/nsMCDLibrary.dll',
@@ -57,13 +58,30 @@ def setUp(self):
                 localfile = os.path.join(tempfile.gettempdir(), 'nsMCDLibrary_Linux32_3.7b.tar.gz')
             if not os.path.exists(localfile):
                 urlretrieve(distantfile, localfile)
-            self.dllname = os.path.join(tempfile.gettempdir(), 'nsMCDLibrary/nsMCDLibrary.so')
-            if not os.path.exists(self.dllname):
+            cls.dllname = os.path.join(tempfile.gettempdir(), 'nsMCDLibrary/nsMCDLibrary.so')
+            if not os.path.exists(cls.dllname):
                 tar = tarfile.open(localfile)
                 tar.extract('nsMCDLibrary/nsMCDLibrary.so', path=tempfile.gettempdir())
         else:
             raise unittest.SkipTest("Not currently supported on OS X")
 
+        # move dll to path discoverable by neuroshare package
+        discoverable_path = pathlib.Path.home() / ".neuroshare" / cls.dllname.split('/')[-1]
+        if not discoverable_path.exists():
+            discoverable_path.parent.mkdir(exist_ok=True)
+            shutil.copyfile(cls.dllname, discoverable_path)
+            cls._cleanup_library = True
+        else:
+            cls._cleanup_library = False
+
+        cls.default_keyword_arguments['dllname'] = cls.dllname
+
+    @classmethod
+    def tearDownClass(cls) -> None:
+        if cls._cleanup_library:
+            discoverable_path = pathlib.Path.home() / ".neuroshare" / cls.dllname.split('/')[-1]
+            discoverable_path.unlink(missing_ok=True)
+
     def test_with_multichannel(self):
         for filename in self.files_to_test:
             filename = self.get_local_path(filename)