def test_fixed_values(self):
fit = Fit(model=Model())
fit.parameters = [
{'symbol': 'D', 'guess': 2},
{'symbol': 'k', 'value': 3, 'prefix': 'kilo'},
{'symbol': 'm', 'value': 4},
]
fit.constants = [
{'symbol': 'a', 'value': 10},
{'symbol': 'b', 'value': 5, 'prefix': 'milli'},
{'symbol': 'c', 'value': 'Avogadro constant'},
]
eq_(fit.fixed_values, (3000, 4, 10, 0.005, 6.022140857e+23))
def test_all_variables(self):
fit = Fit(model=Model())
symbols = ('t', 'u', 'x', 'm', 'D', 'k', 'τ', 'a', 'b')
fit.model.add_symbols(*symbols)
fit.free_variables = ['t', 'u']
fit.independent = {'symbol': 'x'}
fit.parameters = [
{'symbol': 'm', 'guess': 2},
{'symbol': 'D', 'guess': 2},
{'symbol': fit.model.symbol('k'), 'value': 3},
{'symbol': 'τ', 'value': 4},
]
fit.constants = [
{'symbol': 'a'},
{'symbol': 'b'},
]
eq_(fit.all_variables, tuple( fit.model.symbol(s) for s in symbols ))
def test_function_with_more_symbols(self):
fit = Fit()
fit.model = Model()
symbols = ('x', 'a', 'b', 'c', 'd', 'e', 'f')
fit.model.add_symbols(*symbols)
x, a, b, c, d, e, f = fit.model.get_symbols(*symbols)
fit.model.expressions['exp'] = a * f + b * e + x * c + d
fit.expression = 'exp'
fit.independent = {'symbol': 'x'}
fit.parameters = [
{'symbol': 'a', 'guess': 2},
{'symbol': 'b', 'guess': 3},
{'symbol': 'c', 'value': 4, 'prefix': 'kilo'},
{'symbol': 'd', 'value': 5},
]
fit.constants = [
{'symbol': 'e', 'value': 10},
{'symbol': 'f', 'value': 12, 'prefix': 'milli'},
]
eq_(fit.function(2, 4, 7), 8075.048)
