def test_initDataFromFits(file, inputs, results): image = AstroImage.initDataFromFits(file, **inputs) fits = pyfits.open(file) data = fits[inputs["ext"]].data fits.close() verifyParameters(image, results) verifyData(image.data, data)
@pytest.mark.parametrize(("obs","inputs","results"), creation_data) def test_creation(obs,inputs,results): obs = ObservationModule(obs,**inputs) verifyParameters(obs,results) prep_data = [ ( {'input_image':os.path.join(currentdir,"iabf01c5q_flt.fits"),'image_scale':0.3, 'image_wcs':True,'image_ext':1}, AstroImage.initFromFits(os.path.join(currentdir,"iabf01c5q_flt.fits"),ext=1) ), ( {'input_image':os.path.join(currentdir,"iabf01c5q_flt.fits"),'image_scale':1.2, 'image_wcs':False,'image_ext':1}, AstroImage.initDataFromFits(os.path.join(currentdir,"iabf01c5q_flt.fits"),ext=1, scale=[1.2,1.2]) ) ] @pytest.mark.parametrize(("in_img","astro_img"), prep_data) def test_prepImage(in_img,astro_img): obs = ObservationModule({}) obs.prepImage(in_img) test_AstroImage.verifyImage(obs.images[in_img['input_image']],astro_img) test_AstroImage.verifyData(obs.images[in_img['input_image']].data,astro_img.data) obs_data = [ ( #observation {'filters':["F070W","F115W","F150W"], 'offset':[{'offset_ra':0.,'offset_dec':0.,'offset_pa':0.,'offset_centre':False},