In [ ]:

    

from __future__ import print_function
from distutils.version import LooseVersion as Version
import sys



OK = '\x1b[42m[ OK ]\x1b[0m'
FAIL = "\x1b[41m[FAIL]\x1b[0m"

try:
    import importlib
except ImportError:
    print(FAIL, "Python version 3.4 (or 2.7) is required,"
                " but %s is installed." % sys.version)

    
def import_version(pkg, min_ver, fail_msg=""):
    mod = None
    try:
        mod = importlib.import_module(pkg)
        if pkg in {'PIL'}:
            ver = mod.VERSION
        else:
            ver = mod.__version__
        if Version(ver) < min_ver:
            print(FAIL, "%s version %s or higher required, but %s installed."
                  % (lib, min_ver, ver))
        else:
            print(OK, '%s version %s' % (pkg, ver))
    except ImportError:
        print(FAIL, '%s not installed. %s' % (pkg, fail_msg))
    return mod


# first check the python version
print('Using python in', sys.prefix)
print(sys.version)
pyversion = Version(sys.version)
if pyversion >= "3":
    if pyversion < "3.4":
        print(FAIL, "Python version 3.4 (or 2.7) is required,"
                    " but %s is installed." % sys.version)
elif pyversion >= "2":
    if pyversion < "2.7":
        print(FAIL, "Python version 2.7 is required,"
                    " but %s is installed." % sys.version)
else:
    print(FAIL, "Unknown Python version: %s" % sys.version)

print()
requirements = {'numpy': "1.6.1", 'scipy': "0.9", 'matplotlib': "1.0",
                'IPython': "3.0", 'sklearn': "0.18", 'pandas': "0.18",
                'PIL': "1.1.7"}

# now the dependencies
for lib, required_version in list(requirements.items()):
    import_version(lib, required_version)

# graphviz is a bit different
testgraph = """'digraph Tree {\nnode [shape=box] ;\n0 [label="X[0] <= -3.6693\\ngini = 0.6666\\nsamples = 100\\nvalue = [34, 33, 33]"] ;\n1 [label="X[0] <= -8.3538\\ngini = 0.5\\nsamples = 66\\nvalue = [0, 33, 33]"] ;\n0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ;\n2 [label="gini = 0.0\\nsamples = 33\\nvalue = [0, 33, 0]"] ;\n1 -> 2 ;\n3 [label="gini = 0.0\\nsamples = 33\\nvalue = [0, 0, 33]"] ;\n1 -> 3 ;\n4 [label="gini = 0.0\\nsamples = 34\\nvalue = [34, 0, 0]"] ;\n0 -> 4 [labeldistance=2.5, labelangle=-45, headlabel="False"] ;\n}'"""
try:
    import graphviz
    graphviz.Source(testgraph)
    print(OK, 'graphviz version %s' % (graphviz.__version__))
except:
    print("graphviz is not installed."
          "It is not required but you will miss out on some plots."
          "\nYou can install it using 'conda install graphviz python-graphviz'")