add tests for progressbar

spacetelescope · Mar 7, 2024 · ef7f9e9 · ef7f9e9
1 parent 0dd0a85
commit ef7f9e9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/poppy/tests/test_core.py b/poppy/tests/test_core.py
@@ -178,6 +178,9 @@ def test_multiwavelength_opticalsystem():
     assert np.allclose(psf[0].data, output), \
         "Multi-wavelength PSF does not match weighted sum of individual wavelength PSFs"
 
+    # test that it's also possible to display a progress bar for multi wave calculations
+    psf = osys.calc_psf(wavelength=wavelengths, weight=weights, progressbar=True)
+
     return psf
 
 

diff --git a/poppy/tests/test_instrument.py b/poppy/tests/test_instrument.py
@@ -166,7 +166,7 @@ def test_instrument_calc_datacube():
 
     inst = instrument.Instrument()
     psf = inst.calc_datacube(WAVELENGTHS_ARRAY, fov_pixels=FOV_PIXELS,
-                       detector_oversample=2, fft_oversample=2)
+                       detector_oversample=2, fft_oversample=2, progressbar=True)
     assert psf[0].header['NWAVES'] == len(WAVELENGTHS_ARRAY), \
         "Number of wavelengths in PSF header does not match number requested"
     assert len(psf[0].data.shape) == 3, "Incorrect dimensions for output cube"