Add test for _burn_overlay hack

pytroll · May 27, 2020 · 85d9a06 · 85d9a06
1 parent e73308f
commit 85d9a06
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/trollimage/tests/test_image.py b/trollimage/tests/test_image.py
@@ -2004,3 +2004,22 @@ def dummy_fun(pil_obj, *args, **kwargs):
             self.assertEqual(dummy_args, [({}, ), {}])
             res.data.data.compute()
             self.assertEqual(dummy_args, [(OrderedDict(), 'Hey', 'Jude'), {'chorus': "La lala lalalala"}])
+
+        # Test HACK for _burn_overlay
+        dummy_args = [(OrderedDict(), ), {}]
+
+        def _burn_overlay(pil_obj, *args, **kwargs):
+            dummy_args[0] = args
+            dummy_args[1] = kwargs
+            return pil_obj
+
+        img = xrimage.XRImage(data)
+        with mock.patch.object(xrimage, "PILImage") as pi:
+            pil_img = mock.MagicMock()
+            pi.fromarray = mock.Mock(wraps=lambda *args, **kwargs: pil_img)
+            res = img.apply_pil(_burn_overlay, 'RGB')
+            # check that the pil image generation is delayed
+            pi.fromarray.assert_not_called()
+            # make it happen
+            res.data.data.compute()
+            pil_img.convert.assert_called_with('RGB')