def test_create_single_solid_pipeline_with_alias(): a_source = define_stub_solid('A_source', [input_set('A_input')]) stub_solid = define_stub_solid('stub', [{'a_key': 'stubbed_thing'}]) single_solid_pipeline = PipelineDefinition.create_single_solid_pipeline( PipelineDefinition( solids=[a_source, create_root_solid('A')], dependencies={ SolidInstance('A', alias='aliased'): { 'A_input': DependencyDefinition(a_source.name) }, }, ), 'aliased', { 'aliased': { 'A_input': stub_solid, }, }, ) result = execute_pipeline(single_solid_pipeline) assert result.success expected = [{'a_key': 'stubbed_thing'}, {'A': 'transform_called'}] assert result.result_for_solid('aliased').transformed_value() == expected
def test_create_single_solid_pipeline(): stub_solid = define_stub_solid('stub', [{'a key': 'a value'}]) single_solid_pipeline = PipelineDefinition.create_single_solid_pipeline( PipelineDefinition(solids=create_diamond_solids(), dependencies=diamond_deps()), 'A', { 'A': { 'A_input': stub_solid, }, }, ) result = execute_pipeline(single_solid_pipeline) assert result.success
def test_only_final(): pipeline = PipelineDefinition.create_single_solid_pipeline( define_part_fourteen_step_one(), 'final', injected_solids={ 'final': { 'num1': define_stub_solid('stub_a', 3), 'num2': define_stub_solid('stub_b', 4), } }) result = execute_pipeline(pipeline) assert result.success assert len(result.result_list) == 3 assert result.result_for_solid('stub_a').transformed_value() == 3 assert result.result_for_solid('stub_b').transformed_value() == 4 assert result.result_for_solid('final').transformed_value() == 12