Small correction to unit tests for getPSF()

euclidlike/euclidlike_psf.py

@@ -75,7 +75,7 @@ def getPSF(
         ccd, bandpass,
         ccd_pos=None, wcs=None,
         wavelength=None, gsparams=None,
-        logger=None
+        logger=None, stepk = None, maxk = None
 ):
     """Get a single PSF for Euclid like simulation.
 
@@ -130,7 +130,7 @@ def getPSF(
         )
 
     # Now get psf model
-    psf = _get_single_psf_obj(ccd, bandpass, ccd_pos, wavelength, gsparams)
+    psf = _get_single_psf_obj(ccd, bandpass, ccd_pos, wavelength, gsparams, stepk, maxk)
     # Apply WCS.
     # The current version is in arcsec units, but oriented parallel to the
     # image coordinates. So to apply the right WCS, project to pixels using the
@@ -143,7 +143,7 @@ def getPSF(
     return psf
 
 
-def _get_single_psf_obj(ccd, bandpass, ccd_pos, wavelength, gsparams):
+def _get_single_psf_obj(ccd, bandpass, ccd_pos, wavelength, gsparams, stepk = None, maxk = None):
     """
     Routine for making a single PSF.  This gets called by `getPSF` after it
     parses all the options that were passed in.  Users will not directly
@@ -152,7 +152,7 @@ def _get_single_psf_obj(ccd, bandpass, ccd_pos, wavelength, gsparams):
 
     wave_list, im_list = get_euclid_wavelength_psf()
     # instantiate psf object from list of images and wavelengths
-    psf_obj = galsim.InterpolatedChromaticObject.from_images(im_list, wave_list)
+    psf_obj = galsim.InterpolatedChromaticObject.from_images(im_list, wave_list, _force_stepk = stepk, _force_maxk = maxk)
     if wavelength is not None:
         if isinstance(wavelength, galsim.Bandpass):
             wave = wavelength.effective_wavelength

tests/test_psf.py

@@ -57,7 +57,7 @@ def test_get_psf_function():
 
     # creating narrow SED at desired wavelength
     x = np.linspace(euc_bp.blue_limit, euc_bp.red_limit, 500)
-    x[285] =  wave #index 185 roughly corresponds to this wavelength, manually setting
+    x[285] =  wave #index 285 roughly corresponds to this wavelength, manually setting
     f = np.full(len(x), 1e-30)
     f[285] = 100
     lk_table = galsim.LookupTable(x = x, f = f)
@@ -75,7 +75,7 @@ def test_get_psf_function():
     true_img = true_obj.drawImage(euc_bp, scale=scale)
     # images identicial within 0.01% of total flux
     np.testing.assert_allclose(
-        psf_img.array, true_img.array, atol=1e-5*np.sum(true_img.array),
+        psf_img.array, true_img.array, atol=1e-4*np.sum(true_img.array),
         err_msg='getPSF() does replicate image with very narrow SED centered at desired wavelength')
     return