# import
import os
import numpy as np
import matplotlib.pyplot as plt
import WrightTools as wt
from WrightTools import fit
# define
here = os.path.abspath(os.path.dirname(__file__))
# noisey gaussian
xi = np.linspace(-100, 100, 25)
yi = 20*np.exp(-0.5*((xi-5)/20.)**2)
yi = np.random.poisson(yi)
plt.scatter(xi, yi)
# fitted
g = wt.fit.Gaussian()
ps = g.fit(yi, xi)
xi = np.linspace(-100, 100, 101)
model = g.evaluate(ps, xi)
# plot
plt.plot(xi, model)
wt.artists.savefig(os.path.join(here, 'fit_function.png'))