def test_create_prf_flux(): """ Check if create_prf works correctly when fluxes are specified. """ prf = create_prf(image, positions, psf_size, fluxes=fluxes, subsampling=1) assert np.abs(prf._prf_array[0, 0].sum() - 1) < 1E-10 assert_allclose(prf._prf_array[0, 0], test_psf, atol=1E-8)
def test_create_prf_nan(): """ Check if create_prf deals correctly with nan values. """ image_nan = image.copy() image_nan[52, 52] = np.nan image_nan[52, 48] = np.nan prf = create_prf(image_nan, positions, psf_size, subsampling=1) assert not np.isnan(prf._prf_array[0, 0]).any()
def test_create_prf_median(): """ Check if create_prf works correctly on simulated data. """ prf = create_prf(image, positions, psf_size, subsampling=1, mode='median') assert_allclose(prf._prf_array[0, 0], test_psf, atol=1E-8)