def test_aggsed_write_file_exists(self):
sed = Sed(x=self.x,y=self.y,yerr=self.yerr)
spectrum = Spectrum(x=self.x, y=self.y, yerr=self.yerr)
aggsed = AggregateSed([sed, spectrum])
filename = test_directory+'aggsed_file_exists.dat'
aggsed.write(filename)
self.assertEqual(os.path.exists(filename), True)
def test_aggsed_no_counts2(self):
sed = Sed(x=self.x,y=self.y,yerr=self.yerr)
spectrum = Spectrum(x=self.x, y=self.y, yerr=self.yerr)
aggsed = AggregateSed([sed, spectrum])
filename = test_directory+'aggsed_counts2.dat'
aggsed.write(filename)
data = astropy.io.ascii.read(filename)
self.assertEqual(data.colnames, ['x','y','y_err'])
def test_aggsed_counts1(self):
sed = Sed(x=self.x,y=self.y,yerr=self.yerr)
sed.counts = numpy.ones(numpy.array(sed._toarray()[0]).size, dtype=numpy.int_)
spectrum = Spectrum(x=self.x, y=self.y, yerr=self.yerr)
aggsed = AggregateSed([sed, spectrum])
filename = test_directory+'aggsed_counts1.dat'
aggsed.write(filename)
self.assert_(os.path.exists(filename))
data = astropy.io.ascii.read(filename)
self.assertEqual(all(counts == numpy.nan for counts in data['counts']), False)
