Try importing either ruamel.yaml or ruamel_yaml (#61)

exdir/core/attribute.py

@@ -1,8 +1,11 @@
 from enum import Enum
-import ruamel_yaml as yaml
 import os
 import numpy as np
 import exdir
+try:
+    import ruamel_yaml as yaml
+except ImportError:
+    import ruamel.yaml as yaml
 
 def _quote_strings(value):
     if isinstance(value, str):

exdir/core/exdir_object.py

@@ -4,9 +4,12 @@
 
 from enum import Enum
 import os
-import ruamel_yaml as yaml
 import warnings
 import pathlib
+try:
+    import ruamel_yaml as yaml
+except ImportError:
+    import ruamel.yaml as yaml
 
 import exdir
 

exdir/core/group.py

@@ -1,9 +1,12 @@
 import os
 import re
-import ruamel_yaml as yaml
 import pathlib
 import numpy as np
 import exdir
+try:
+    import ruamel_yaml as yaml
+except ImportError:
+    import ruamel.yaml as yaml
 try:
     from collections import abc
 except ImportError:

tests/test_help_functions.py

@@ -3,10 +3,13 @@
 import os
 import pathlib
 import six
-import ruamel_yaml as yaml
 import quantities as pq
 import numpy as np
 import pytest
+try:
+    import ruamel_yaml as yaml
+except ImportError:
+    import ruamel.yaml as yaml
 
 import exdir
 import exdir.core
@@ -217,4 +220,3 @@ def test_open_object(setup_teardown_file):
     loaded_grp = exob.open_object(path)
 
     assert grp2 == loaded_grp
-

tests/test_object.py

@@ -14,9 +14,12 @@
 
 import pytest
 import os
-import ruamel_yaml as yaml
 import pathlib
 import exdir
+try:
+    import ruamel_yaml as yaml
+except ImportError:
+    import ruamel.yaml as yaml
 
 from exdir.core import Object, Attribute
 # TODO Remove this import and use import <> as <> instead