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)