source: sasmodels/sasmodels/model_test.py @ d6adfbe

# -*- coding: utf-8 -*-
"""
Run model unit tests.

Usage::

     python -m sasmodels.model_test [opencl|dll|opencl_and_dll] model1 model2 ...

     if model1 is 'all', then all except the remaining models will be tested

Each model is tested using the default parameters at q=0.1, (qx,qy)=(0.1,0.1),
and the ER and VR are computed.  The return values at these points are not
considered.  The test is only to verify that the models run to completion,
and do not produce inf or NaN.

Tests are defined with the *tests* attribute in the model.py file.  *tests*
is a list of individual tests to run, where each test consists of the
parameter values for the test, the q-values and the expected results.  For
the effective radius test, the q-value should be 'ER'.  For the VR test,
the q-value should be 'VR'.  For 1-D tests, either specify the q value or
a list of q-values, and the corresponding I(q) value, or list of I(q) values.

That is::

    tests = [
        [ {parameters}, q, I(q)],
        [ {parameters}, [q], [I(q)] ],
        [ {parameters}, [q1, q2, ...], [I(q1), I(q2), ...]],

        [ {parameters}, (qx, qy), I(qx, Iqy)],
        [ {parameters}, [(qx1, qy1), (qx2, qy2), ...],
                        [I(qx1,qy1), I(qx2,qy2), ...]],

        [ {parameters}, 'ER', ER(pars) ],
        [ {parameters}, 'VR', VR(pars) ],
        ...
    ]

Parameters are *key:value* pairs, where key is one of the parameters of the
model and value is the value to use for the test.  Any parameters not given
in the parameter list will take on the default parameter value.

Precision defaults to 5 digits (relative).
"""

import sys
import unittest

import numpy as np

from .core import list_models, load_model_definition
from .core import load_model_cl, load_model_dll
from .core import make_kernel, call_kernel, call_ER, call_VR

def annotate_exception(exc, msg):
    """
    Add an annotation to the current exception, which can then be forwarded
    to the caller using a bare "raise" statement to reraise the annotated
    exception.
    Example::
        >>> D = {}
        >>> try: 
        ...    print D['hello']
        ... except Exception,exc: 
        ...    annotate_exception(exc, "while accessing 'D'")
        ...    raise
        Traceback (most recent call last):
            ...
        KeyError: "hello while accessing 'D'"
    """
    args = exc.args
    if not args:
        exc.args = (msg,)
    else:
        try:
            arg0 = " ".join((args[0],msg))
            exc.args = tuple([arg0] + list(args[1:]))
        except:
            exc.args = (" ".join((str(exc),msg)),)
    
def suite(loaders, models):

    ModelTestCase = _hide_model_case_from_nosetests()
    suite = unittest.TestSuite()

    if models[0] == 'all':
        skip = models[1:]
        models = list_models()
    else:
        skip = []
    for model_name in models:
        if model_name in skip: continue
        model_definition = load_model_definition(model_name)

        smoke_tests = [
            [{},0.1,None],
            [{},(0.1,0.1),None],
            [{},'ER',None],
            [{},'VR',None],
            ]
        tests = smoke_tests + getattr(model_definition, 'tests', [])
        
        if tests: # in case there are no smoke tests...
            #print '------'
            #print 'found tests in', model_name
            #print '------'

            # if ispy then use the dll loader to call pykernel
            # don't try to call cl kernel since it will not be
            # available in some environmentes.
            ispy = callable(getattr(model_definition,'Iq', None))

            # test using opencl if desired
            if not ispy and ('opencl' in loaders and load_model_cl):
                test_name = "Model: %s, Kernel: OpenCL"%model_name
                test = ModelTestCase(test_name, model_definition,
                                     load_model_cl, tests)
                #print "defining", test_name
                suite.addTest(test)

            # test using dll if desired
            if ispy or ('dll' in loaders and load_model_dll):
                test_name = "Model: %s, Kernel: dll"%model_name
                test = ModelTestCase(test_name, model_definition,
                                     load_model_dll, tests)
                #print "defining", test_name
                suite.addTest(test)

    return suite

def _hide_model_case_from_nosetests():
    class ModelTestCase(unittest.TestCase):
        def __init__(self, test_name, definition, loader, tests):
            unittest.TestCase.__init__(self)

            self.test_name = test_name
            self.definition = definition
            self.loader = loader
            self.tests = tests

        def runTest(self):
            try:
                model = self.loader(self.definition)
                for test in self.tests:
                    pars, Q, I = test

                    if not isinstance(I, list):
                        I = [I]
                    if not isinstance(Q, list):
                        Q = [Q]

                    self.assertEqual(len(I), len(Q))

                    if Q[0] == 'ER':
                        Iq = [call_ER(kernel, pars)]
                    elif Q[0] == 'VR':
                        Iq = [call_VR(kernel, pars)]
                    elif isinstance(Q[0], tuple):
                        Qx,Qy = zip(*Q)
                        Q_vectors = [np.array(Qx), np.array(Qy)]
                        kernel = make_kernel(model, Q_vectors)
                        Iq = call_kernel(kernel, pars)
                    else:
                        Q_vectors = [np.array(Q)]
                        kernel = make_kernel(model, Q_vectors)
                        Iq = call_kernel(kernel, pars)

                    self.assertGreater(len(Iq), 0)
                    self.assertEqual(len(I), len(Iq))

                    for q, i, iq in zip(Q, I, Iq):
                        if i is None:
                            # smoke test --- make sure it runs and produces a value
                            self.assertTrue(np.isfinite(iq), 'q:%s; not finite; actual:%s' % (q, iq))
                        else:
                            err = abs(i - iq)
                            nrm = abs(i)
                            self.assertLess(err * 10**5, nrm, 'q:%s; expected:%s; actual:%s' % (q, i, iq))

            except Exception,exc:
                annotate_exception(exc, self.test_name)
                raise

    return ModelTestCase


# let nosetests sniff out the tests
def model_tests():
    tests = suite(['opencl','dll'],['all'])
    for test_i in tests:
        yield test_i.runTest

def main():
    models = sys.argv[1:]
    if models and models[0] == 'opencl':
        if load_model_cl is None:
            print >>sys.stderr, "opencl is not available"
            return 1
        loaders = ['opencl']
        models = models[1:]
    elif models and models[0] == 'dll':
        # TODO: test if compiler is available?
        loaders = ['dll']
        models = models[1:]
    elif models and models[0] == 'opencl_and_dll':
        if load_model_cl is None:
            print >>sys.stderr, "opencl is not available"
            return 1
        loaders = ['opencl', 'dll']
        models = models[1:]
    else:
        loaders = ['opencl', 'dll']
    if not models:
        print >>sys.stderr, "usage: python -m sasmodels.model_test [opencl|dll|opencl_and_dll] model1 model2 ..."
        print >>sys.stderr, "if model1 is 'all', then all except the remaining models will be tested"
        return 1

    #run_tests(loaders, models)
    runner = unittest.TextTestRunner()
    result = runner.run(suite(loaders, models))
    return 1 if result.failures or result.errors else 0

if __name__ == "__main__":
    sys.exit(main())