def test_simple_linear_background_with_single_peak(self):
x = np.linspace(0, 25, 2500)
y_data = gaussian(x, 10, 3, 0.1)
y_bkg = x * 0.4 + 5.0
y_measurement = y_data + y_bkg
y_extracted_bkg = extract_background(x, y_measurement, 1)
self.assertAlmostEqual(np.sum(y_data - (y_measurement - y_extracted_bkg)), 0)
def test_simple_linear_background_with_multiple_peaks(self):
x = np.linspace(0, 24, 2500)
y_data = np.zeros(x.shape)
peaks = [
[10, 3, 0.05],
[12, 6, 0.05],
[12, 9, 0.05],
]
for peak in peaks:
y_data += gaussian(x, peak[0], peak[1], peak[2])
y_bkg = x * 0.4 + 5.0
y_measurement = y_data + y_bkg
y_extracted_bkg = extract_background(x, y_measurement, 0.3)
self.assertAlmostEqual(np.sum(y_data - (y_measurement - y_extracted_bkg)), 0)