def test_initFromFits(file, inputs, results): image = AstroImage.initFromFits(file, **inputs) fits = pyfits.open(file) data = fits[inputs["ext"]].data fits.close() verifyParameters(image, results) verifyData(image.data, data)
'in_path':'/path/to/somewhere','cat_path':'/path/to/somewhere/else', 'out_path':'/path/to/somewhere/else','poisson':True,'flat':False, 'dark':False,'cosmic':False,'version':"9.5"} ) ] @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)