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