Example #1
0
	def setUp(self):
		self.stack = Stacky()
		self.fits = pyfits.HDUList(pyfits.PrimaryHDU(np.arange(4096).reshape(64,64)))
		self.onesfits = pyfits.HDUList(pyfits.PrimaryHDU(np.ones(4096).reshape(64,64)))
		self.nanfits = pyfits.HDUList(pyfits.PrimaryHDU(np.ones(4096).reshape(64,64)))
		self.nanfits[0].data[0:63][0:63] = np.nan
		for fits in [self.fits, self.onesfits, self.nanfits]:
			fits[0].header.update('BUNIT', 'TestUnits', '')
			fits[0].header.update('CRPIX1', 1, '')
			fits[0].header.update('CRPIX2', 1, '')
			fits[0].header.update('CDELT1', -20.0/3600.0, '')
			fits[0].header.update('CDELT2', 20.0/3600.0, '')
			fits[0].header.update('CRVAL1', -(32*self.fits[0].header['CDELT1']), '')
			fits[0].header.update('CRVAL2', -(32*self.fits[0].header['CDELT2']), '')
			fits[0].header.update('CTYPE1', 'RA-CAR', '')
			fits[0].header.update('CTYPE2', 'DEC-CAR', '')
Example #2
0
class TestStampyStacky(unittest.TestCase):
	def setUp(self):
		self.stack = Stacky()
		self.fits = pyfits.HDUList(pyfits.PrimaryHDU(np.arange(4096).reshape(64,64)))
		self.onesfits = pyfits.HDUList(pyfits.PrimaryHDU(np.ones(4096).reshape(64,64)))
		self.nanfits = pyfits.HDUList(pyfits.PrimaryHDU(np.ones(4096).reshape(64,64)))
		self.nanfits[0].data[0:63][0:63] = np.nan
		for fits in [self.fits, self.onesfits, self.nanfits]:
			fits[0].header.update('BUNIT', 'TestUnits', '')
			fits[0].header.update('CRPIX1', 1, '')
			fits[0].header.update('CRPIX2', 1, '')
			fits[0].header.update('CDELT1', -20.0/3600.0, '')
			fits[0].header.update('CDELT2', 20.0/3600.0, '')
			fits[0].header.update('CRVAL1', -(32*self.fits[0].header['CDELT1']), '')
			fits[0].header.update('CRVAL2', -(32*self.fits[0].header['CDELT2']), '')
			fits[0].header.update('CTYPE1', 'RA-CAR', '')
			fits[0].header.update('CTYPE2', 'DEC-CAR', '')
	def testHeaderValues(self):
		location = (0,0)
		stamp = Stampy(12, self.onesfits, location, name="testname")
		self.assertEqual(stamp.header["units"], "TestUnits")
		self.assertEqual(stamp.header["source"], "testname")
		self.assertEqual(stamp.header["radec"], location)
		self.assertEqual(stamp.header["bmean"], 1)
		self.assertEqual(stamp.header["bmedian"], 1)
	def testFlag(self):
		location = (0, 0)
		badlocation = (64,64)
		stamp = Stampy(12, self.onesfits, location, maxnoise=2)
		self.assert_(not stamp.header["flag"]["highnoise"])
		self.assert_(not stamp.header["flag"]["hasnans"])
		self.assert_(not stamp.header["flag"]["dimensionerror"])
		self.assert_(not stamp.header["flag"]["nearsource"])
		self.assertEqual(stamp.header["flag"]["errorflag"], 'a')
		stamp = Stampy(12, self.fits, location, maxnoise=0)
		self.assert_(stamp.header["flag"]["highnoise"])
		self.assert_(not stamp.header["flag"]["hasnans"])
		self.assert_(not stamp.header["flag"]["dimensionerror"])
		self.assert_(not stamp.header["flag"]["nearsource"])
		self.assertEqual(stamp.header["flag"]["errorflag"], 'n')
		stamp = Stampy(66, self.fits, location)
		self.assert_(not stamp.header["flag"]["highnoise"])
		self.assert_(not stamp.header["flag"]["hasnans"])
		self.assert_(stamp.header["flag"]["dimensionerror"])
		self.assert_(not stamp.header["flag"]["nearsource"])
		self.assertEqual(stamp.header["flag"]["errorflag"], 'd')
		stamp = Stampy(12, self.fits, badlocation)
		self.assert_(not stamp.header["flag"]["highnoise"])
		self.assert_(not stamp.header["flag"]["hasnans"])
		self.assert_(stamp.header["flag"]["dimensionerror"])
		self.assert_(not stamp.header["flag"]["nearsource"])
		self.assertEqual(stamp.header["flag"]["errorflag"], 'd')
		stamp = Stampy(12, self.nanfits, location)
		self.assert_(not stamp.header["flag"]["highnoise"])
		self.assert_(stamp.header["flag"]["hasnans"])
		self.assert_(not stamp.header["flag"]["dimensionerror"])
		self.assert_(not stamp.header["flag"]["nearsource"])
		self.assertEqual(stamp.header["flag"]["errorflag"], 'b')

	def testStackReject(self):
		location = (0,0)
		nanstamp = Stampy(12, self.nanfits, location)
		goodstamp = Stampy(12, self.onesfits, location, maxnoise=2)
		noisestamp = Stampy(12, self.onesfits, location, maxnoise=0)
		dimenstamp = Stampy(66, self.onesfits, location, maxnoise=0)
		self.assertEqual(self.stack.add_stamp(goodstamp), 'a')
		self.assertEqual(self.stack.add_stamp(noisestamp), 'n')
		self.assertEqual(self.stack.add_stamp(dimenstamp), 'd')
		self.assertEqual(self.stack.add_stamp(nanstamp), 'b')