def test_pickle_by_evaluation(function): f = function f2 = loads(dumps(f)) mus = parameters_of_type(f.parameter_type, 47) for arg in function_argument(f, 10, 42): mu = next(mus) assert np.all(f.evaluate(arg, mu) == f2.evaluate(arg, mu))
def test_evaluate(function): f = function mus = parameters_of_type(f.parameter_type, 4711) for count in [0, 1, 5, (0, 1), (2, 2, 2)]: arg = function_argument(f, count, 454) result = f.evaluate(arg, next(mus)) assert result.shape == arg.shape[:-1] + f.shape_range