check_examples.py

#!/usr/bin/env python
import sys
import os

sys.dont_write_bytecode = True

ROOT = os.path.abspath(os.path.dirname(__file__))
CLI = "%s %s/bin/refl1d_cli.py %%s %%s" % (sys.executable, ROOT)
EXAMPLEDIR = os.path.join(ROOT, "doc", "examples")

# Add packages to the python path
# Use in-place build of refl1d (requires "python setup.py build_ext --inplace")
packages = [ROOT]
## Add the build dir(s) to the system path
##from distutils.util import get_platform
##platform = '.%s-%s'%(get_platform(), sys.version[:3])
##buildpath = os.path.abspath(os.path.join(ROOT, 'build', 'lib'+platform))
##packages = [buildpath]
try:
    import bumps
except ImportError:
    packages.append(os.path.abspath(os.path.join(ROOT, "..", "bumps")))
try:
    import periodictable
except ImportError:
    packages.append(os.path.abspath(os.path.join(ROOT, "..", "periodictable")))
if "PYTHONPATH" in os.environ:
    packages.append(os.environ["PYTHONPATH"])
os.environ["PYTHONPATH"] = os.pathsep.join(packages)


class Commands(object):
    @staticmethod
    def preview(f):
        return os.system(CLI % (f, "--preview --seed=1"))

    @staticmethod
    def edit(f):
        return os.system(CLI % (f, "--edit --seed=1"))

    @staticmethod
    def chisq(f):
        return os.system(CLI % (f, "--chisq --seed=1"))

    @staticmethod
    def time(f):
        ## Note: use --parallel to check serialization for MPMapper
        # return os.system(CLI%(f, '--time_model --seed=1 --steps=24 --parallel'))
        return os.system(CLI % (f, "--time_model --seed=1 --steps=20"))


examples = [
    "distribution/dist-example.py",
    "ex1/nifilm.py",
    "ex1/nifilm-fit.py",
    "ex1/nifilm-data.py",
    "ex1/nifilm-tof.py",
    "freemag/pmf.py",
    "ill_posed/anticor.py",
    "ill_posed/tethered.py",
    "interface/model.py",
    "mixed/mixed.py",
    "mixed/mixed_magnetic.py",
    # "peaks/model.py",
    "polymer/tethered.py",
    "polymer/freeform.py",
    "profile/model.py",
    "spinvalve/n101G.py",
    "spinvalve/n101G.sta",
    "staj/De2_VATR.py",
    "superlattice/freeform.py",
    "superlattice/NiTi.py",
    "superlattice/PEMU.py",
    "thick/nifilm.py",
    "TOF/du53.py",
    "xray/mlayer-lin.staj",
    "xray/model.py",
    "xray/staj.py",
]


def main():
    if len(sys.argv) == 1 or not hasattr(Commands, sys.argv[1][2:]):
        print("usage: check_examples.py [--preview|--edit|--chisq|--time]")
    else:
        command = getattr(Commands, sys.argv[1][2:])
        for f in examples:
            print("\n" + f)
            status = command(os.path.join(EXAMPLEDIR, f))
            if status != 0:
                sys.exit(1)


if __name__ == "__main__":
    main()