diff --git a/jwst/wfss_contam/tests/test_observations.py b/jwst/wfss_contam/tests/test_observations.py index 71228f6479..82d3cdbfac 100644 --- a/jwst/wfss_contam/tests/test_observations.py +++ b/jwst/wfss_contam/tests/test_observations.py @@ -92,21 +92,19 @@ def test_disperse_oversample_same_result(grism_wcs, segmentation_map): wmin, wmax = np.min(sens_waves), np.max(sens_waves) sens_resp = np.ones(100) seg_wcs = segmentation_map.meta.wcs - 0, (300, 500), 2, False, xoffset = 2200 yoffset = 1000 - xs, ys, areas, lams_out, counts_1, ID = dispersed_pixel( x0, y0, width, height, lams, flxs, order, wmin, wmax, sens_waves, sens_resp, seg_wcs, grism_wcs, ID, naxis, oversample_factor=1, extrapolate_sed=False, xoffset=xoffset, yoffset=yoffset) - + xs, ys, areas, lams_out, counts_3, ID = dispersed_pixel( x0, y0, width, height, lams, flxs, order, wmin, wmax, sens_waves, sens_resp, seg_wcs, grism_wcs, ID, naxis, oversample_factor=3, extrapolate_sed=False, xoffset=xoffset, yoffset=yoffset) - assert np.isclose(np.sum(counts_1), np.sum(counts_3), rtol=1/sens_waves.size) \ No newline at end of file + assert np.isclose(np.sum(counts_1), np.sum(counts_3), rtol=1/sens_waves.size)