Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
lambda *args, **kwargs: models.ExpressionModel(self.expression, *args, **kwargs))
super().setUp()
def setUp(self):
self.true_values = lambda: dict(re=5, im=5)
self.guess = lambda: dict(re=2, im=2)
self.model_constructor = models.ComplexConstantModel
super().setUp()
def setUp(self):
self.true_values = lambda: dict(c3=2, c2=5, c1=2, c0=8)
self.guess = lambda: dict(c3=1, c1=1, c2=6, c0=3)
self.model_constructor = models.PolynomialModel
self.args = (3,)
super().setUp()
def factory_gaussian(center=0.1, sigma=0.1, amplitude=1):
"""Return an lmfit Gaussian model that can be used to fit data.
Arguments are initial values for the model parameters.
Returns
An `lmfit.Model` object with all the parameters already initialized.
"""
model = lmfit.models.GaussianModel()
model.set_param_hint('center', value=center, min=-1, max=2)
model.set_param_hint('sigma', value=sigma)
model.set_param_hint('amplitude', value=amplitude)
return model
'lm_save_modelresult': save_modelresult,
'lm_load_modelresult': load_modelresult,
}
for name in ('BreitWignerModel', 'ComplexConstantModel',
'ConstantModel', 'DampedHarmonicOscillatorModel',
'DampedOscillatorModel', 'DonaichModel',
'ExponentialGaussianModel', 'ExponentialModel',
'ExpressionModel', 'GaussianModel', 'Interpreter',
'LinearModel', 'LognormalModel', 'LorentzianModel',
'MoffatModel', 'ParabolicModel', 'Pearson7Model',
'PolynomialModel', 'PowerLawModel',
'PseudoVoigtModel', 'QuadraticModel',
'RectangleModel', 'SkewedGaussianModel',
'StepModel', 'StudentsTModel', 'VoigtModel'):
val = getattr(lmfit.models, name, None)
if val is not None:
exports[name] = val
_larch_builtins = {'_math': exports}
if comp in curmodels:
curmodels.remove(comp)
prefix = curmodels[0]
label = "%s(prefix='%s')" % (model, prefix)
title = "%s: %s " % (prefix[:-1], model)
title = prefix[:-1]
mclass_kws = {'prefix': prefix}
if 'step' in model.lower():
form = model.lower().replace('step', '').strip()
if form.startswith('err'):
form = 'erf'
label = "Step(form='%s', prefix='%s')" % (form, prefix)
title = "%s: Step %s" % (prefix[:-1], form[:3])
mclass = lm_models.StepModel
mclass_kws['form'] = form
minst = mclass(form=form, prefix=prefix)
else:
if model in ModelFuncs:
mclass = getattr(lm_models, ModelFuncs[model])
else:
mclass = getattr(lm_models, model+'Model')
minst = mclass(prefix=prefix)
panel = GridPanel(self.mod_nb, ncols=2, nrows=5, pad=1, itemstyle=CEN)
def SLabel(label, size=(80, -1), **kws):
return SimpleText(panel, label,
size=size, style=wx.ALIGN_LEFT, **kws)
usebox = Check(panel, default=True, label='Use in Fit?', size=(100, -1))
for comp in self.fit_components:
if comp in curmodels:
curmodels.remove(comp)
prefix = curmodels[0]
label = "%s(prefix='%s')" % (model, prefix)
title = "%s: %s" % (prefix[:-1], (model+' '*8)[:8])
mclass_kws = {'prefix': prefix}
if 'step' in model.lower():
form = model.lower().replace('step', '').strip()
if form.startswith('err'): form = 'erf'
label = "Step(form='%s', prefix='%s')" % (form, prefix)
title = "%s: Step %s" % (prefix[:-1], form[:3])
mclass = lm_models.StepModel
mclass_kws['form'] = form
minst = mclass(form=form, prefix=prefix)
else:
mclass = getattr(lm_models, model+'Model')
minst = mclass(prefix=prefix)
panel = GridPanel(self.mod_nb, ncols=1, nrows=1, pad=1, itemstyle=CEN)
def SLabel(label, size=(80, -1), **kws):
return SimpleText(panel, label,
size=size, style=wx.ALIGN_LEFT, **kws)
usebox = Check(panel, default=True, label='Use?', size=(75, -1))
delbtn = Button(panel, 'Delete Model', size=(120, -1),
action=partial(self.onDeleteComponent, prefix=prefix))
pick2msg = SimpleText(panel, " ", size=(75, -1))
pick2btn = Button(panel, 'Pick Data Range', size=(135, -1),
title = prefix[:-1]
mclass_kws = {'prefix': prefix}
if 'step' in model.lower():
form = model.lower().replace('step', '').strip()
if form.startswith('err'):
form = 'erf'
label = "Step(form='%s', prefix='%s')" % (form, prefix)
title = "%s: Step %s" % (prefix[:-1], form[:3])
mclass = lm_models.StepModel
mclass_kws['form'] = form
minst = mclass(form=form, prefix=prefix)
else:
if model in ModelFuncs:
mclass = getattr(lm_models, ModelFuncs[model])
else:
mclass = getattr(lm_models, model+'Model')
minst = mclass(prefix=prefix)
panel = GridPanel(self.mod_nb, ncols=2, nrows=5, pad=1, itemstyle=CEN)
def SLabel(label, size=(80, -1), **kws):
return SimpleText(panel, label,
size=size, style=wx.ALIGN_LEFT, **kws)
usebox = Check(panel, default=True, label='Use in Fit?', size=(100, -1))
bkgbox = Check(panel, default=False, label='Is Baseline?', size=(125, -1))
if isbkg:
bkgbox.SetValue(1)
delbtn = Button(panel, 'Delete This Component', size=(200, -1),
action=partial(self.onDeleteComponent, prefix=prefix))