Esempio n. 1
0
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},