Add optional limit_output switch to xarray bilinear resampling

Author: Zaczero

satpy/resample/kdtree.py

@@ -214,7 +214,7 @@ def __init__(self, source_geo_def, target_geo_def):
         self.resampler = None
 
     def precompute(self, mask=None, radius_of_influence=50000, epsilon=0,
-                   reduce_data=True, cache_dir=False, **kwargs):
+                   reduce_data=True, limit_output=True, cache_dir=False, **kwargs):
         """Create bilinear coefficients and store them for later use."""
         try:
             from pyresample.bilinear import XArrayBilinearResampler
@@ -230,7 +230,8 @@ def precompute(self, mask=None, radius_of_influence=50000, epsilon=0,
                           radius_of_influence=radius_of_influence,
                           neighbours=32,
                           epsilon=epsilon,
-                          reduce_data=reduce_data)
+                          reduce_data=reduce_data,
+                          limit_output=limit_output)
 
             self.resampler = XArrayBilinearResampler(**kwargs)
             try:

satpy/tests/test_resample.py

@@ -338,20 +338,26 @@ class TestBilinearResampler(unittest.TestCase):
     """Test the bilinear resampler."""
 
     @mock.patch("pyresample.bilinear.XArrayBilinearResampler")
-    def test_precompute_forwards_reduce_data(self, xr_resampler):
-        """Test that precompute forwards reduce_data to pyresample bilinear."""
+    def test_precompute_forwards_bilinear_options(self, xr_resampler):
+        """Test that precompute forwards bilinear options to pyresample."""
         from satpy.resample.kdtree import BilinearResampler
 
         _, source_area, _, _, target_area = get_test_data()
         resampler = BilinearResampler(source_area, target_area)
         resampler.precompute()
         assert xr_resampler.call_args.kwargs["reduce_data"] is True
+        assert xr_resampler.call_args.kwargs["limit_output"] is True
 
         xr_resampler.reset_mock()
         resampler = BilinearResampler(source_area, target_area)
         resampler.precompute(reduce_data=False)
         assert xr_resampler.call_args.kwargs["reduce_data"] is False
 
+        xr_resampler.reset_mock()
+        resampler = BilinearResampler(source_area, target_area)
+        resampler.precompute(limit_output=False)
+        assert xr_resampler.call_args.kwargs["limit_output"] is False
+
     @mock.patch("satpy.resample.kdtree._move_existing_caches")
     @mock.patch("satpy.resample.kdtree.BilinearResampler._create_cache_filename")
     @mock.patch("pyresample.bilinear.XArrayBilinearResampler")