Exemple #1
0
def test_observation_get_has(image_data):
    obs = Observation(image=image_data['image'])
    assert np.all(obs.image == image_data['image'])
    assert not obs.has_bmask()
    assert not obs.has_ormask()
    assert not obs.has_noise()
    assert not obs.has_psf()
    assert not obs.has_psf_gmix()
    assert not obs.has_gmix()

    obs = Observation(image=image_data['image'], weight=image_data['weight'])
    assert np.all(obs.weight == image_data['weight'])

    obs = Observation(image=image_data['image'], bmask=image_data['bmask'])
    assert np.all(obs.bmask == image_data['bmask'])
    assert obs.has_bmask()

    obs = Observation(image=image_data['image'], ormask=image_data['ormask'])
    assert np.all(obs.ormask == image_data['ormask'])
    assert obs.has_ormask()

    obs = Observation(image=image_data['image'], noise=image_data['noise'])
    assert np.all(obs.noise == image_data['noise'])
    assert obs.has_noise()

    obs = Observation(image=image_data['image'], psf=image_data['psf'])
    assert np.all(obs.psf.image == image_data['psf'].image)
    assert obs.has_psf()
    assert not obs.has_psf_gmix()

    obs = Observation(image=image_data['image'], gmix=image_data['gmix'])
    assert np.all(
        obs.gmix.get_full_pars() == image_data['gmix'].get_full_pars())
    assert obs.has_gmix()

    obs = Observation(image=image_data['image'], psf=image_data['psf_gmix'])
    assert np.all(obs.psf.image == image_data['psf_gmix'].image)
    assert obs.has_psf()
    assert obs.has_psf_gmix()
    assert np.all(
        obs.psf.gmix.get_full_pars() == obs.get_psf_gmix().get_full_pars())

    obs = Observation(image=image_data['image'],
                      jacobian=image_data['jacobian'])
    assert (obs.jacobian.get_galsim_wcs() ==
            image_data['jacobian'].get_galsim_wcs())

    obs = Observation(image=image_data['image'], meta=image_data['meta'])
    assert obs.meta == image_data['meta']