def test_with_show_plot():
wav = np.arange(5e3, 5080., 0.02)
def spec(x):
return np.exp(-(x - 5040.)**2/1000.)
polynomial_fit_values = bf.blazeFit(wav, spec(wav), 5e-2)#, showplot=True)
assert len(polynomial_fit_values) == 8
def test_with_noise():
wav = np.arange(5e3, 5080., 0.02)
def spec(x):
normal_noise = np.random.normal(loc=0, scale=0.1, size=len(x))
blaze_function = np.exp(-(x - 5040.)**2/1000.)
return blaze_function + normal_noise
polynomial_fit_values = bf.blazeFit(wav, spec(wav), 5e-2, verbose=True)
assert len(polynomial_fit_values) == 8
def test_blazeFit():
"""
Ensure that blazeFit returns a 7th order polynomial
"""
wav = np.arange(5e3, 5080., 0.02)
def spec(x):
return np.exp(-(x - 5040.)**2/1000.)
polynomial_fit_values = bf.blazeFit(wav, spec(wav), 5e-2)
assert len(polynomial_fit_values) == 8
