fix(Python 2): use a few aliases for exceptions

* FileNotFoundError = IOError * ModuleNotFoundError = ImportError
modflowpy · Oct 11, 2019 · 97d22fe · 97d22fe
1 parent 7dc1ec8
commit 97d22fe
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/autotest/t011_test.py b/autotest/t011_test.py
@@ -4,10 +4,14 @@
 """
 
 import os
+import sys
 import flopy
 import numpy as np
 from nose.tools import raises
 
+if sys.version_info[0] == 2:
+    FileNotFoundError = IOError
+
 
 def test_mflistfile():
     pth = os.path.join('..', 'examples', 'data', 'freyberg')

diff --git a/flopy/utils/gridintersect.py b/flopy/utils/gridintersect.py
@@ -1,11 +1,14 @@
 import numpy as np
+import sys
+if sys.version_info[0] == 2:
+    ModuleNotFoundError = ImportError
 from .geometry import transform
 try:
     from shapely.geometry import MultiPoint, Point, Polygon, box
     from shapely.strtree import STRtree
     from shapely.affinity import translate, rotate
 except ModuleNotFoundError:
-    print("Shapely is needed for grid intersect operations!"
+    print("Shapely is needed for grid intersect operations! "
           "Please install shapely.")
 
 

diff --git a/flopy/utils/mflistfile.py b/flopy/utils/mflistfile.py
@@ -15,6 +15,9 @@
 
 from ..utils.utils_def import totim_to_datetime
 
+if sys.version_info[0] == 2:
+    FileNotFoundError = IOError
+
 
 class ListBudget(object):
     """