more stage tests

api/tests/chain/test_blend_mask.py

@@ -0,0 +1,23 @@
+import unittest
+from PIL import Image
+
+from onnx_web.chain.blend_mask import BlendMaskStage
+from onnx_web.params import HighresParams, UpscaleParams
+
+
+class BlendMaskStageTests(unittest.TestCase):
+    def test_empty(self):
+        stage = BlendMaskStage()
+        sources = []
+        result = stage.run(
+            None,
+            None,
+            None,
+            None,
+            sources,
+            highres=HighresParams(False, 1, 0, 0),
+            upscale=UpscaleParams(""),
+            stage_mask=Image.new("RGBA", (64, 64)),
+        )
+
+        self.assertEqual(len(result), 0)

api/tests/chain/test_reduce_crop.py

@@ -0,0 +1,23 @@
+import unittest
+
+from onnx_web.chain.reduce_crop import ReduceCropStage
+from onnx_web.params import HighresParams, Size, UpscaleParams
+
+
+class ReduceCropStageTests(unittest.TestCase):
+    def test_empty(self):
+        stage = ReduceCropStage()
+        sources = []
+        result = stage.run(
+            None,
+            None,
+            None,
+            None,
+            sources,
+            highres=HighresParams(False, 1, 0, 0),
+            upscale=UpscaleParams(""),
+            origin=Size(0, 0),
+            size=Size(128, 128),
+        )
+
+        self.assertEqual(len(result), 0)

api/tests/chain/test_upscale_highres.py

@@ -10,4 +10,4 @@ class UpscaleHighresStageTests(unittest.TestCase):
         sources = []
         result = stage.run(None, None, None, None, sources, highres=HighresParams(False,1, 0, 0), upscale=UpscaleParams(""))
 
-        self.assertEqual(len(result), 0)
+        self.assertEqual(len(result), 0)