comp_utest.py 文件源码

def make_func(varname, wrfnc, timeidx, method, squeeze, meta):
    def func(self):

        try:
            args = get_args(varname, wrfnc, timeidx, method, squeeze)
        except ProjectionError: # Don't fail for this
            return

        routine = ROUTINE_MAP[varname]

        kwargs = {"meta" : meta}
        result = routine(*args, **kwargs)

        ref = getvar(wrfnc, varname, timeidx, method, squeeze, cache=None, 
                     meta=meta)

        nt.assert_allclose(to_np(result), to_np(ref))

        if meta:
            self.assertEqual(result.dims, ref.dims)

    return func