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)
