def test_remove_segment(self):
segment1 = Spectrum(x = numpy.arange(1000,10000,10),
y = numpy.arange(1000,10000,10),
yerr=numpy.arange(1000,10000,10)*.01,
z=1.0)
segment2 = Sed(x=numpy.arange(1000,10000,500),
y=numpy.arange(1000,10000,500),
yerr=numpy.arange(1000,10000,500)*.01)
segment3 = Sed(x=numpy.arange(1000,10000,500),
y=numpy.arange(1000,10000,500),
yerr=numpy.arange(1000,10000,500)*.01,
z = 0.35)
segment4 = Spectrum(x = numpy.arange(1000,10000,10),
y = numpy.arange(1000,10000,10),
yerr=numpy.arange(1000,10000,10)*.01,
z=1.0)
aggsed = Stack([segment1, segment2, segment3, segment4])
aggsed.remove_segment(segment1)
self.assertEqual(len(aggsed), 3)
self.assertEqual(len(aggsed.x[0]), len(numpy.arange(1000,10000,500)))
