def test_sed_write_with_counts(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_) filename = test_directory+'sed_counts.dat' sed.write(filename) self.assertEqual(os.path.exists(filename), True) data = astropy.io.ascii.read(filename) self.assertEqual(data['counts'][3],1.0)
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)