Merge pull request #80 from dstansby/astropy-v-check

Check astropy version when importing pfsspy
dstansby · Feb 18, 2019 · 77151ba · 77151ba
2 parents 0d4d7a1 + 58fa8b5
commit 77151ba
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pfsspy/__init__.py b/pfsspy/__init__.py
@@ -1,3 +1,6 @@
+import distutils.version
+
+import astropy
 import astropy.coordinates as coord
 import astropy.constants as const
 import astropy.units as u
@@ -14,6 +17,12 @@
 __version__ = get_versions()['version']
 del get_versions
 
+# Do a version check for astropy
+if (distutils.version.LooseVersion(astropy.__version__) <
+        distutils.version.LooseVersion("3")):
+    raise RuntimeError('pfsspy requires astropy v3 to run ' +
+                       f'(found version {astropy.__version__} installed)')
+
 
 class Grid:
     """