fix tests

This commit is contained in:
2026-03-29 18:23:07 -07:00
parent d94e92666d
commit 2b17a2594f
5 changed files with 27 additions and 6 deletions

View File

@@ -14,7 +14,7 @@ def test_threshold_mask():
ThresholdMask._broadcast_fn = lambda nid, uri: previews.append(uri)
ThresholdMask._current_node_id = "test"
mask, = node.process(field, method="absolute", threshold=0.5, direction="above")
mask, table = node.process(field, method="absolute", threshold=0.5, direction="above")
assert mask.dtype == np.uint8
assert mask.shape == (64, 64)
assert np.all(mask[:, :32] == 0)