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