def test_solid_def():
@lambda_solid
def produce_string():
return "foo"
@solid(
input_defs=[InputDefinition("input_one", String)],
output_defs=[OutputDefinition(Any)],
config_schema={"another_field": Int},
)
def solid_one(_context, input_one):
raise Exception("should not execute")
pipeline_def = PipelineDefinition(
solid_defs=[produce_string, solid_one],
dependencies={"solid_one": {"input_one": DependencyDefinition("produce_string")}},
)
assert len(pipeline_def.solids[0].output_handles()) == 1
assert isinstance(pipeline_def.solid_named("solid_one"), Solid)
solid_one_solid = pipeline_def.solid_named("solid_one")
assert solid_one_solid.has_input("input_one")
assert isinstance(solid_one_solid.input_def_named("input_one"), InputDefinition)
assert len(solid_one_solid.input_dict) == 1
assert len(solid_one_solid.output_dict) == 1
assert str(solid_one_solid.input_handle("input_one")) == (
"SolidInputHandle(input_name=\"'input_one'\", solid_name=\"'solid_one'\")"
)
assert repr(solid_one_solid.input_handle("input_one")) == (
"SolidInputHandle(input_name=\"'input_one'\", solid_name=\"'solid_one'\")"
)
assert str(solid_one_solid.output_handle("result")) == (
"SolidOutputHandle(output_name=\"'result'\", solid_name=\"'solid_one'\")"
)
assert repr(solid_one_solid.output_handle("result")) == (
"SolidOutputHandle(output_name=\"'result'\", solid_name=\"'solid_one'\")"
)
assert solid_one_solid.output_handle("result") == SolidOutputHandle(
solid_one_solid, solid_one_solid.output_dict["result"]
)
assert (
len(pipeline_def.dependency_structure.input_to_upstream_outputs_for_solid("solid_one")) == 1
)
assert (
len(
pipeline_def.dependency_structure.output_to_downstream_inputs_for_solid(
"produce_string"
)
)
== 1
)
assert len(pipeline_def.dependency_structure.input_handles()) == 1
assert len(pipeline_def.dependency_structure.items()) == 1
def test_solid_def():
@lambda_solid
def produce_string():
return 'foo'
@solid(
input_defs=[InputDefinition('input_one', String)],
output_defs=[OutputDefinition(Any)],
config={'another_field': Int},
)
def solid_one(_context, input_one):
raise Exception('should not execute')
pipeline_def = PipelineDefinition(
solid_defs=[produce_string, solid_one],
dependencies={'solid_one': {'input_one': DependencyDefinition('produce_string')}},
)
assert len(pipeline_def.solids[0].output_handles()) == 1
assert isinstance(pipeline_def.solid_named('solid_one'), Solid)
solid_one_solid = pipeline_def.solid_named('solid_one')
assert solid_one_solid.has_input('input_one')
assert isinstance(solid_one_solid.input_def_named('input_one'), InputDefinition)
assert len(solid_one_solid.input_dict) == 1
assert len(solid_one_solid.output_dict) == 1
assert str(solid_one_solid.input_handle('input_one')) == (
'SolidInputHandle(definition_name="\'solid_one\'", input_name="\'input_one\'", '
'solid_name="\'solid_one\'")'
)
assert repr(solid_one_solid.input_handle('input_one')) == (
'SolidInputHandle(definition_name="\'solid_one\'", input_name="\'input_one\'", '
'solid_name="\'solid_one\'")'
)
assert str(solid_one_solid.output_handle('result')) == (
'SolidOutputHandle(definition_name="\'solid_one\'", output_name="\'result\'", '
'solid_name="\'solid_one\'")'
)
assert repr(solid_one_solid.output_handle('result')) == (
'SolidOutputHandle(definition_name="\'solid_one\'", output_name="\'result\'", '
'solid_name="\'solid_one\'")'
)
assert solid_one_solid.output_handle('result') == SolidOutputHandle(
solid_one_solid, solid_one_solid.output_dict['result']
)
assert (
len(pipeline_def.dependency_structure.input_to_upstream_outputs_for_solid('solid_one')) == 1
)
assert (
len(
pipeline_def.dependency_structure.output_to_downstream_inputs_for_solid(
'produce_string'
)
)
== 1
)
assert len(pipeline_def.dependency_structure.input_handles()) == 1
assert len(pipeline_def.dependency_structure.items()) == 1
def test_solid_def():
@lambda_solid
def produce_string():
return 'foo'
@solid(
inputs=[InputDefinition('input_one', types.String)],
outputs=[OutputDefinition(types.Any)],
config_field=Field(Dict({'another_field': Field(types.Int)})),
)
def solid_one(_context, input_one):
raise Exception('should not execute')
pipeline_def = PipelineDefinition(
solids=[produce_string, solid_one],
dependencies={
'solid_one': {
'input_one': DependencyDefinition('produce_string')
}
},
)
assert len(pipeline_def.solids[0].output_handles()) == 1
assert isinstance(pipeline_def.solid_named('solid_one'), Solid)
solid_one_solid = pipeline_def.solid_named('solid_one')
assert solid_one_solid.has_input('input_one')
assert isinstance(solid_one_solid.input_def_named('input_one'),
InputDefinition)
assert len(solid_one_solid.input_defs) == 1
assert len(solid_one_solid.output_defs) == 1
assert str(solid_one_solid.input_handle('input_one')) == (
'SolidInputHandle(definition_name="\'solid_one\'", input_name="\'input_one\'", '
'solid_name="\'solid_one\'")')
assert repr(solid_one_solid.input_handle('input_one')) == (
'SolidInputHandle(definition_name="\'solid_one\'", input_name="\'input_one\'", '
'solid_name="\'solid_one\'")')
assert str(solid_one_solid.output_handle('result')) == (
'SolidOutputHandle(definition_name="\'solid_one\'", output_name="\'result\'", '
'solid_name="\'solid_one\'")')
assert repr(solid_one_solid.output_handle('result')) == (
'SolidOutputHandle(definition_name="\'solid_one\'", output_name="\'result\'", '
'solid_name="\'solid_one\'")')
assert solid_one_solid.output_handle('result') == SolidOutputHandle(
solid_one_solid, solid_one_solid.output_defs[0])
assert len(
pipeline_def.dependency_structure.deps_of_solid_with_input(
'solid_one')) == 1
assert len(
pipeline_def.dependency_structure.depended_by_of_solid(
'produce_string')) == 1
assert len(pipeline_def.dependency_structure.input_handles()) == 1
assert len(pipeline_def.dependency_structure.items()) == 1
