def test_from_header(): fname = 'tests/test_files/1904-66_SIN.fits' header = fits.getheader(fname) helper = WCSHelper.from_header(header) if helper.beam is None: raise AssertionError() del header['BMAJ'], header['BMIN'], header['BPA'] helper = WCSHelper.from_header(header) if helper.beam is not None: raise AssertionError()
def test_from_header(): """Test that we can make a beam from a fitsheader""" fname = 'tests/test_files/1904-66_SIN.fits' header = fits.getheader(fname) helper = WCSHelper.from_header(header) if helper.beam is None: raise AssertionError() del header['BMAJ'], header['BMIN'], header['BPA'] helper = WCSHelper.from_header(header) if helper.beam is not None: raise AssertionError()
def test_from_header(): """Test that we can make a beam from a fitsheader""" fname = 'tests/test_files/1904-66_SIN.fits' header = fits.getheader(fname) helper = WCSHelper.from_header(header) if helper.beam is None: raise AssertionError() del header['BMAJ'], header['BMIN'], header['BPA'] # Raise an error when the beam information can't be determined try: _ = WCSHelper.from_header(header) except AssertionError as e: pass else: raise AssertionError( "Header with no beam information should thrown an exception.") return
def test_galactic_coords(): """Test that we can work with galactic coordinates""" fname = 'tests/test_files/1904-66_SIN.lb.fits' header = fits.getheader(fname) try: helper = WCSHelper.from_header(header) except ValueError as e: raise AssertionError("galactic coordinates break WCSHelper") return
src.island = 102 src.peak_flux = 80 * noise cat.append(src) return cat if __name__ == "__main__": # configure logging logging.basicConfig(format="%(module)s:%(levelname)s %(message)s") log = logging.getLogger("Aegean") logging_level = logging.DEBUG # if options.debug else logging.INFO log.setLevel(logging_level) image, wcs = make_noise_image() header = wcs.to_header() header['BMAJ'] = psf[0] / 3600 header['BMIN'] = psf[1] / 3600 header['BPA'] = psf[2] hdu = fits.HDUList(hdus=[fits.PrimaryHDU(header=header, data=image)]) header = hdu[0].header cat = make_catalogue() wcshelper = WCSHelper.from_header(header) hdu[0].data += AeRes.make_model(cat, image.shape, wcshelper) hdu.writeto('synthetic_test.fits', overwrite=True) psf_hdu = make_psf_map(hdu) psf_hdu.writeto('synthetic_test_psf.fits', overwrite=True)