def test_dynamic_output_solid():
@solid(output_defs=[DynamicOutputDefinition()])
def should_work(_):
yield DynamicOutput(1, mapping_key="1")
yield DynamicOutput(2, mapping_key="2")
result = execute_in_process(should_work)
assert result.success
assert result.output_values["result"]["1"] == 1
assert result.output_values["result"]["2"] == 2
def test_execute_solid_nonexistent_inputs():
emit_one, add = get_solids()
with pytest.raises(
DagsterInvalidDefinitionError,
match=re.escape(
'Invalid dependencies: solid "emit_one" does not have input "x". Available inputs: []'
),
):
execute_in_process(emit_one, input_values={"x": 5})
with pytest.raises(
DagsterInvalidDefinitionError,
match=re.escape(
'Invalid dependencies: solid "add" does not have input "z". '
"Available inputs: ['x', 'y']"
),
):
execute_in_process(add, input_values={"z": 5})
def test_execute_solid_with_inputs():
@solid
def add_one(_, x):
return 1 + x
result = execute_in_process(add_one,
input_values={"x": 5},
output_capturing_enabled=True)
assert result.success
assert result.output_values["result"] == 6
def test_execute_graph_with_inputs():
emit_one, add = get_solids()
@graph
def add_one(x):
return add(x, emit_one())
result = execute_in_process(add_one, input_values={"x": 5})
assert result.success
assert result.output_values["result"] == 6
assert result.result_for_node("emit_one").output_values["result"] == 1
def test_basic_graph():
emit_one, add = get_solids()
@graph
def get_two():
return add(emit_one(), emit_one())
assert isinstance(get_two, GraphDefinition)
result = execute_in_process(get_two)
assert result.success
def test_execute_graph():
emit_one, add = get_solids()
@graph
def emit_two():
return add(emit_one(), emit_one())
@graph
def emit_three():
return add(emit_two(), emit_one())
result = execute_in_process(emit_three)
assert result.success
def test_nothing_in():
@op(out=Out(dagster_type=Nothing))
def noop():
pass
@op(ins={"after": In(dagster_type=Nothing)})
def on_complete():
return "cool"
@graph
def nothing_test():
on_complete(noop())
result = execute_in_process(nothing_test)
assert result.success
def test_execute_graph():
emit_one, add = get_solids()
@graph
def emit_two():
return add(emit_one(), emit_one())
@graph
def emit_three():
return add(emit_two(), emit_one())
result = execute_in_process(emit_three, output_capturing_enabled=True)
assert result.success
assert result.output_values["result"] == 3
assert result.result_for_node("add").output_values["result"] == 3
assert result.result_for_node("emit_two").output_values["result"] == 2
assert result.result_for_node("emit_one").output_values["result"] == 1
assert (result.result_for_node("emit_two").result_for_node(
"emit_one").output_values["result"] == 1)
assert (result.result_for_node("emit_two").result_for_node(
"emit_one_2").output_values["result"] == 1)
def test_op_config():
@op(config_schema={"conf_str": str})
def my_op(context):
assert context.op_config == {"conf_str": "foo"}
my_op(build_op_context(config={"conf_str": "foo"}))
@graph
def basic():
my_op()
result = execute_in_process(basic,
run_config={
"ops": {
"basic": {
"ops": {
"my_op": {
"config": {
"conf_str": "foo"
}
}
}
}
}
})
assert result.success
result = basic.to_job(config={
"ops": {
"my_op": {
"config": {
"conf_str": "foo"
}
}
}
}).execute_in_process()
assert result.success
def test_execute_solid():
emit_one, _ = get_solids()
result = execute_in_process(emit_one)
assert result.success