Exemple #1
0
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)