コード例 #1
0
ファイル: test_iris_sed.py プロジェクト: jbudynk/sedstacker
    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)