Ensure "empty" output pixels come out as 0, not nan, for "mean" moasic

astropy · Dec 8, 2023 · 059aeb5 · 059aeb5
1 parent e46fbd7
commit 059aeb5
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 1 deletion.
diff --git a/reproject/mosaicking/coadd.py b/reproject/mosaicking/coadd.py
@@ -276,6 +276,7 @@ def reproject_and_coadd(
         if combine_function == "mean":
             with np.errstate(invalid="ignore"):
                 output_array /= output_footprint
+                output_array[output_footprint == 0] = 0
 
     elif combine_function in ("first", "last", "min", "max"):
         for array in arrays:

diff --git a/reproject/mosaicking/tests/reference/test_coadd_solar_map.fits b/reproject/mosaicking/tests/reference/test_coadd_solar_map.fits
diff --git a/reproject/mosaicking/tests/test_coadd.py b/reproject/mosaicking/tests/test_coadd.py
@@ -236,7 +236,7 @@ def test_coadd_background_matching_one_array(self, reproject_function):
         np.testing.assert_allclose(array, array_matched)
         np.testing.assert_allclose(footprint, footprint_matched)
 
-    @pytest.mark.parametrize("combine_function", ["first", "last", "min", "max", "sum"])
+    @pytest.mark.parametrize("combine_function", ["first", "last", "min", "max", "sum", "mean"])
     @pytest.mark.parametrize("match_background", [True, False])
     def test_footprint_correct(self, reproject_function, combine_function, match_background):
         # Test that the output array is zero outside the returned footprint