FIX: Ensure dependencies installed

saketkc · saketkc · commit 081bd4d9fdf9 · 2015-03-11T15:06:06.000-07:00
diff --git a/setup.py b/setup.py
@@ -114,6 +114,19 @@ def configuration(parent_package='', top_path=None):
 
     return config
 
+def is_scipy_installed():
+    try:
+        import scipy
+    except ImportError:
+        return False
+    return True
+
+def is_numpy_installed():
+    try:
+        import numpy
+    except ImportError:
+        return False
+    return True
 
 def setup_package():
     metadata = dict(name=DISTNAME,
@@ -162,6 +175,18 @@ def setup_package():
 
         metadata['version'] = VERSION
     else:
+        if is_numpy_installed() is False:
+            sys.stderr.write("Error: Numerical Python(NumPy) is not installed.\n"
+                             "scikit-learn requires NumPy.\n"
+                             "Instruction installations are available on scikit-learn website:\
+                             http://scikit-learn.org/stable/install.html\n")
+            sys.exit(1)
+        if is_scipy_installed() is False:
+            sys.stderr.write("Error: Scientific Python(SciPy) is not installed.\n"
+                             "scikit-learn requires SciPy.\n"
+                             "Instruction installations are available on scikit-learn website:\
+                             http://scikit-learn.org/stable/install.html\n")
+            sys.exit(1)
         from numpy.distutils.core import setup
 
         metadata['configuration'] = configuration
