def test_sed_write_file_exists(self):
sed = Sed(x=self.x,y=self.y,yerr=self.yerr)
filename = test_directory+'sed_file_exists.dat'
sed.write(filename)
self.assertEqual(os.path.exists(filename), True)
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)
