def testSaveStacks(self):
genstack.save_noise = self.savenoisepos
genstack.stack_build(self.fits, 12, 0.05, 0.05, 1, astWCS.WCS(self.fits[0].header, mode="pyfits"), None, False, 1)
genstack.save_noise = self.savenoiseneg
genstack.stack_build(self.fits, 12, 0.04, -0.05, 1, astWCS.WCS(self.fits[0].header, mode="pyfits"), None, False, 1)
genstack.units = "TestUnits1"
genstack.save_stacks("test")
meanfile = pyfits.open("test_mean.fits")
medianfile = pyfits.open("test_median.fits")
self.assertEqual(meanfile[0].data[6,6], 1)
self.assertEqual(meanfile[0].header["BUNIT"], "TestUnits1")
self.assertEqual(medianfile[0].header["BUNIT"], "TestUnits1")
self.assertEqual(meanfile[0].header["CDELT1"], -20.0/3600.0)
self.assertEqual(medianfile[0].header["CDELT2"], 20.0/3600.0)
self.assertEqual(meanfile[0].header["CRVAL1"], 120.0/3600.0)
self.assertEqual(medianfile[0].header["CRVAL2"], -120.0/3600.0)
os.remove("test_mean.fits")
os.remove("test_median.fits")
def testNameBuild(self): genstack.save_noise = self.savenoisepos genstack.stack_build(self.fits, 12, 0.05, 0.05, 1, astWCS.WCS(self.fits[0].header, mode="pyfits"), None, False, 1) genstack.save_noise = self.savenoiseneg genstack.stack_build(self.fits, 12, 0.04, -0.05, 1, astWCS.WCS(self.fits[0].header, mode="pyfits"), None, False, 1)
