Spaces:
Running
Running
File size: 780 Bytes
d29f839 fbf786e d29f839 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from lynxkite.core import ops, workspace
from lynxkite.core.executors import simple
async def test_optional_inputs():
@ops.op("test", "one")
def one():
return 1
@ops.op("test", "maybe add")
def maybe_add(a: int, b: int | None = None):
return a + (b or 0)
assert maybe_add.__op__.inputs == [
ops.Input(name="a", type=int, position="left"),
ops.Input(name="b", type=int | None, position="left"),
]
simple.register("test")
ws = workspace.Workspace(env="test", nodes=[], edges=[])
a = ws.add_node(one)
b = ws.add_node(maybe_add)
await ws.execute()
assert b.data.error == "Missing input: a"
ws.add_edge(a, "output", b, "a")
outputs = await ws.execute()
assert outputs[b.id, "output"] == 1
|