def test_norm_at_point_spectrum3(self):
spectrum = Spectrum(x = range(0, 101),
y = range(0, 101))
norm_spectrum = spectrum.normalize_at_point(20, 50, dx=50)
control_avg_flux = 35.0
control_norm_constant = 50.0/control_avg_flux
self.assertEqual(norm_spectrum.norm_constant, control_norm_constant)
def test_norm_at_point_spectrum1(self):
spectrum = Spectrum(x = numpy.linspace(3000.0, 10000.0, num=10000),
y = numpy.linspace(1e-13, 1e-11, num=10000))
norm_spectrum = spectrum.normalize_at_point(5000.0, 1e-13)
assert abs(norm_spectrum.norm_constant - 0.034145) < 0.00001
self.assertEqual(len(norm_spectrum.yerr), len(norm_spectrum.x))
self.assert_(numpy.isnan(norm_spectrum.yerr[1]))
def test_norm_at_point_spectrum2(self):
spectrum = Spectrum(x = range(0, 101),
y = range(0, 101))
norm_spectrum = spectrum.normalize_at_point(20, 50, dx=10)
control_norm_constant = 50.0/20.0
control_norm_spectrum_y = spectrum.y*control_norm_constant
self.assertEqual(norm_spectrum.norm_constant, 2.5)
numpy.testing.assert_array_almost_equal(norm_spectrum.y, control_norm_spectrum_y)
