def test_background_out_of_range_throws_error(self): x1 = np.linspace(0, 10) x2 = np.linspace(-10, -1) spec = Spectrum(x1, x1) background_spectrum = Spectrum(x2, x2) with self.assertRaises(BkgNotInRangeError): spec.background_spectrum = background_spectrum
def test_using_background_spectrum(self): x = np.linspace(-5, 5, 100) spec_y = x ** 2 bkg_y = x spec = Spectrum(x, spec_y) background_spectrum = Spectrum(x, bkg_y) spec.background_spectrum = background_spectrum new_x, new_y = spec.data self.array_almost_equal(new_x, x) self.array_almost_equal(new_y, spec_y - bkg_y)
def test_using_background_spectrum_with_different_spacing(self): x = np.linspace(-5, 5, 100) spec_y = x ** 2 x_bkg = np.linspace(-5, 5, 99) bkg_y = x_bkg spec = Spectrum(x, spec_y) background_spectrum = Spectrum(x_bkg, bkg_y) spec.background_spectrum = background_spectrum new_x, new_y = spec.data self.array_almost_equal(new_x, x) self.array_almost_equal(new_y, spec_y - x)