def test_outside_norm_ranges(self): sed1 = IrisSed(x=[1,2,3,4,5], y=[1,2,3,4,5], id='sed1') sed2 = IrisSed(x=[10,20,30,40], y=[1,2,3,4], id='sed2') stack = IrisStack([sed1, sed2]) norm_stack = stack.normalize_by_int(minWavelength=2, maxWavelength=9) self.assertEqual(norm_stack.excluded[0], 'sed2') sed3 = IrisSed(x=[20,30,40], y=[1,2,3], id='sed3') stack.add_segment(sed3) norm_stack = stack.normalize_by_int(minWavelength=2, maxWavelength=9) numpy.testing.assert_array_equal(norm_stack.excluded, ['sed2', 'sed3']) self.assertEqual(len(norm_stack), 3)