# -*- coding: utf-8 -*-
import lmfit
def test_wrap_function():
get_names = lambda p: [p_key for p_key in p ]
def func(A, b, c, d=5, e=10):
return A + b + c + d
x0 = [1, 2, 3]
para, f = lmfit.make_paras_and_func(func, x0)
assert(get_names(para) == ['A', 'b', 'c'])
y1 = f(para)
y2 = func(*x0)
assert(y1==y2)
x0 = [1, 2, 3, 4]
para, f = lmfit.make_paras_and_func(func, x0)
assert(get_names(para) == ['A', 'b', 'c', 'd'])
y1 = f(para)
y2 = func(*x0)
assert(y1==y2)
x0 = [1, 2, 3]
para, f = lmfit.make_paras_and_func(func, x0, {'e': 3})
assert(get_names(para) == ['A', 'b', 'c', 'e'])
y1 = f(para)
y2 = func(*x0)
assert(y1==y2)