def test_complex_comp_solid_definition():
@solid
def return_one(_):
return 1
@solid
def take_many(_, items):
return items
@composite_solid
def comp_solid(this_number):
take_many(
[return_one(), this_number,
return_one.alias("return_one_also")()])
comp_solid_meta = build_composite_solid_def_snap(comp_solid)
assert isinstance(comp_solid_meta, CompositeSolidDefSnap)
assert (deserialize_json_to_dagster_namedtuple(
serialize_dagster_namedtuple(comp_solid_meta)) == comp_solid_meta)
index = DependencyStructureIndex(comp_solid_meta.dep_structure_snapshot)
assert index.get_invocation("return_one")
assert index.get_invocation("take_many")
assert index.get_upstream_outputs("take_many",
"items")[0].solid_name == "return_one"
assert index.get_upstream_outputs(
"take_many", "items")[1].solid_name == "return_one_also"
def test_basic_comp_solid_definition():
@solid
def return_one(_):
return 1
@solid
def take_one(_, one):
return one
@composite_solid
def comp_solid():
take_one(return_one())
comp_solid_meta = build_composite_solid_def_snap(comp_solid)
assert isinstance(comp_solid_meta, CompositeSolidDefSnap)
assert (deserialize_json_to_dagster_namedtuple(
serialize_dagster_namedtuple(comp_solid_meta)) == comp_solid_meta)
index = DependencyStructureIndex(comp_solid_meta.dep_structure_snapshot)
assert index.get_invocation("return_one")
assert index.get_invocation("take_one")
assert index.get_upstream_output("take_one",
"one").solid_name == "return_one"
assert index.get_upstream_output("take_one", "one").output_name == "result"
def test_noop_comp_solid_definition():
@solid
def noop_solid(_):
pass
@composite_solid
def comp_solid():
noop_solid()
comp_solid_meta = build_composite_solid_def_snap(comp_solid)
assert isinstance(comp_solid_meta, CompositeSolidDefSnap)
assert (deserialize_json_to_dagster_namedtuple(
serialize_dagster_namedtuple(comp_solid_meta)) == comp_solid_meta)