matplotlib Code

#!/usr/bin/env python
"""Minimal test script to check for modules needed in python course"""

modules = ['numpy','scipy','matplotlib','IPython','mayavi','pyvtk','Image',
           'visual']

for mname in modules:
    try:
        exec "import %s" % mname
    except ImportError:
        print '*** ERROR: module %s could not be imported.' % mname
    else:
        print '%s: OK' % mname

print 'Also remember to check that SPE is installed.'