def test_normalize_at_point_avg_mult(self): sed1 = IrisSed(x=self.x,y=self.y,yerr=self.yerr) sed2 = IrisSed(x=numpy.array([2,4,5,8,10]), y=numpy.arange(5)+1.0, yerr=numpy.arange(5)+1.0*0.1) y = numpy.array([5.0, 15.0, 7.0, 4.5, 13.5, 10.5]) x = numpy.array([0.5, 1.5, 3.0, 5.5, 10.5, 21.0]) sed3 = IrisSed(x=x, y=y, yerr=y*0.1) stack = IrisStack([sed1, sed2, sed3]) # normalize SEDs with avg statistic norm_stack = stack.normalize_at_point(5.0, 1.0, stats='avg', norm_operator=0) numpy.testing.assert_array_almost_equal(norm_stack[0].y, (8/3.)/0.5*sed1.y) numpy.testing.assert_array_almost_equal(norm_stack[1].y, (8/3.)/3.0*sed2.y) self.assertAlmostEqual(norm_stack[2].norm_constant, (8/3.)/4.5)