feat: gate mask_from_stash (paint or zeros)

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

tests/test_gate.py

@@ -1,4 +1,9 @@
 # tests/test_gate.py
+import io
+
+import torch
+from PIL import Image
+
 from gates import gate
 
 def test_route_tuple_places_image_at_chosen():
@@ -9,3 +14,14 @@ def test_route_tuple_places_image_at_chosen():
 def test_route_tuple_length_is_max():
     B = object()
     assert len(gate.route_tuple(0, "IMG", B, max_routes=10)) == 10
+
+def test_mask_from_stash_none_is_zeros():
+    img = torch.zeros((1, 6, 4, 3))
+    m = gate.mask_from_stash(None, img)
+    assert m.shape == (1, 6, 4) and float(m.max()) == 0.0
+
+def test_mask_from_stash_decodes_png():
+    buf = io.BytesIO(); Image.new("L", (4, 6), 255).save(buf, "PNG")
+    img = torch.zeros((1, 6, 4, 3))
+    m = gate.mask_from_stash(buf.getvalue(), img)
+    assert m.shape == (1, 6, 4) and float(m.min()) > 0.99