checkenv.py

# Check that the packages are installed.
from pkgutil import iter_modules
import sys


def check_import(packagename):
    if packagename in (name for _, name, _ in iter_modules()):
        return True
    else:
        return False

assert sys.version_info.major >= 3 and sys.version_info.minor >= 6, 'Please install Python 3.6!'

packages = ['jupyter', 'pymc3', 'seaborn', 'matplotlib', 'numpy', 'scipy',
            'pandas', 'tqdm', 'jupyterlab']

all_passed = True

for p in packages:
    assert check_import(p),\
        '{0} not present. Please install via pip or conda.'.format(p)

if all_passed:
    print('All checks passed. Your environment is good to go!')