Decorators for numpy-to-matlab regression automation

rt_insert.py

import decorator

class RuntimeInsertionManager(object):
    def insert_with_args(self, *args, **dargs):
        @decorator.decorator
        def _wrapper(wrapped, *_args, **_dargs):
            if hasattr(self, 'callback'):
                return self.callback(wrapped, _args, _dargs, args, dargs)
            return wrapped(*_args, **_dargs)
        return _wrapper

MANAGER = RuntimeInsertionMananger()
mimic = MANAGER.insert_with_args

test_whatever.py

from rt_insert import MANAGER
from whatever_ported import whatever_ported

try:
    import matlab.engine as engine
except ImportError:
    try:
        import oct2py
        engine = oct2py.Oct2Py(convert_to_float=False)
    except ImportError:
        raise ImportError("Found neither 'matlab.engine' nor 'oct2py'")

engine.addpath('./')

def callback(func, func_args, func_dargs, dec_args, dec_dargs):
    mfunc_name = dec_args[0]
    mfunc_result = getattr(engine, mfunc_name)(*func_args)
    local_result = func(*func_args, **func_dargs)
    assert type(mfunc_result) == type(local_result), \
        "octave type={} != numpy type={}".format(type(mfunc_result), type(local_result))
    if type(mfunc_result) == type(local_result) == type(np.array([])):
        assert mfunc_result.dtype == local_result.dtype, \
            "octave dtype={} != numpy dtype={}".format(mfunc_result.dtype, local_result.dtype)
        assert mfunc_result.shape == local_result.shape, \
            "octave shape={} != numpy shape={}".format(mfunc_result.shape, local_result.shape)
    assert np.allclose(mfunc_result, local_result)
    return local_result

MANAGER.callback = callback

TEST_CASES = [(1, 2), (3, 4), (5, 6)]

def test_whatever_ported():
    for a, b in TEST_CASES:
        yield whatever_ported, a, b

whatever.m

function c = whatever(a, b)
  c = a+b

whatever_ported.py

from rt_insert import mimic

@mimic('whatever')
def whatever_ported(a, b):
    return a+b