Merge pull request #1317 from zm711/rawio-tests

Add test for get_rawio_class

Author: apdavison

neo/test/rawiotest/test_get_rawio.py

@@ -0,0 +1,27 @@
+from neo.rawio import get_rawio_class
+from pathlib import Path
+from tempfile import TemporaryDirectory
+
+
+def test_get_rawio_class():
+    # use plexon io suffix for testing here
+    non_existant_file = Path('non_existant_folder/non_existant_file.plx')
+    non_existant_file.unlink(missing_ok=True)
+    ios = get_rawio_class(non_existant_file)
+
+    assert ios
+
+    # cleanup
+    non_existant_file.unlink(missing_ok=True)
+
+
+def test_get_rawio_class_nonsupported_rawio():
+
+    non_existant_file = Path('non_existant_folder/non_existant_file.fake')
+    non_existant_file.unlink(missing_ok=True)
+    ios = get_rawio_class(non_existant_file)
+
+    assert ios is None
+
+    # cleanup
+    non_existant_file.unlink(missing_ok=True)