check_env.py

#!/usr/bin/env python

"""Computational environment validation script for the 2014 SciPy Conference Tutorial:

  Reproducible Research: Walking the Walk.

https://github.com/reproducible-research/scipy-tutorial-2014
"""

import sys
import subprocess


return_value = 0

required_packages = ['numpy', 'scipy', 'matplotlib', 'SimpleITK']
for package in required_packages:
    print('Importing ' + package + ' ...')
    try:
        __import__(package, globals(), locals(), [], 0)
    except ImportError:
        print('Error: could not import ' + package)
        return_value += 1


print('')
required_executables = ['git',
    'dexy',
    'ipython',
    'latex',
    'nosetests',
    'mplayer']
for executable in required_executables:
    print('Executing ' + executable + ' ...')
    try:
        process = subprocess.Popen([executable, '--help'],
                                          stderr=subprocess.STDOUT,
                                          stdout=subprocess.PIPE)
        process.wait()
    except OSError:
        print('Error: could not execute ' + executable)
        return_value += 1


if return_value is 0:
    print('\nSuccess.')
else:
    print('\nA defect was found in your environment, please see the messages ' +
          'above.')
sys.exit(return_value)