Shim for FileNotFoundError in tests

gdx/__init__.py

@@ -1,6 +1,6 @@
 # coding: utf-8
-from __future__ import absolute_import, division, print_function, \
-    unicode_literals
+from __future__ import (absolute_import, division, print_function,
+                        unicode_literals)
 from itertools import cycle
 from logging import debug, info
 # commented: for debugging

gdx/api.py

@@ -1,8 +1,9 @@
 # coding: utf-8
-from __future__ import absolute_import, division, print_function, \
-    unicode_literals
+from __future__ import (absolute_import, division, print_function,
+                        unicode_literals)
 from os.path import dirname
 import sys
+from .pycompat import FileNotFoundError
 
 PY3 = sys.version_info[0] >= 3
 
@@ -14,9 +15,6 @@
     from future import standard_library
     standard_library.install_aliases()
 
-    class FileNotFoundError(OSError):
-        pass
-
 import gdxcc
 
 

gdx/pycompat.py

@@ -0,0 +1,11 @@
+import sys
+
+PY3 = sys.version_info[0] >= 3
+
+if PY3:
+    from builtins import FileNotFoundError
+else:
+    class FileNotFoundError(OSError):
+        pass
+
+__all__ = ['FileNotFoundError']

gdx/test/test_gdx.py

@@ -1,8 +1,9 @@
 from collections import OrderedDict
 
+import numpy as np
 import pytest
 
-import numpy as np
+from gdx.pycompat import FileNotFoundError
 
 
 @pytest.fixture(scope='session')