Python define_multi_mode_with_resources_pipeline Examples

Example #1

def test_execute_multi_mode_with_resources():
    pipeline_def = define_multi_mode_with_resources_pipeline()

    add_mode_result = execute_pipeline(
        pipeline=pipeline_def,
        mode="add_mode",
        run_config={"resources": {
            "op": {
                "config": 2
            }
        }},
    )

    assert add_mode_result.result_for_solid(
        "apply_to_three").output_value() == 5

    mult_mode_result = execute_pipeline(
        pipeline=pipeline_def,
        mode="mult_mode",
        run_config={"resources": {
            "op": {
                "config": 3
            }
        }},
    )

    assert mult_mode_result.result_for_solid(
        "apply_to_three").output_value() == 9

Example #2

File: test_mode_definitions.py Project: varokas/dagster-1

def test_execute_multi_mode_with_resources():
    pipeline_def = define_multi_mode_with_resources_pipeline()

    add_mode_result = execute_pipeline(
        pipeline=pipeline_def,
        mode='add_mode',
        run_config={'resources': {
            'op': {
                'config': 2
            }
        }},
    )

    assert add_mode_result.result_for_solid(
        'apply_to_three').output_value() == 5

    mult_mode_result = execute_pipeline(
        pipeline=pipeline_def,
        mode='mult_mode',
        run_config={'resources': {
            'op': {
                'config': 3
            }
        }},
    )

    assert mult_mode_result.result_for_solid(
        'apply_to_three').output_value() == 9

Example #3

File: test_presets.py Project: zuik/dagster

def test_preset_yaml_roundtrip():
    pipeline = define_multi_mode_with_resources_pipeline()

    preset_def = pipeline.get_preset("add")

    with open(
            file_relative_path(
                __file__,
                "../environments/multi_mode_with_resources/add_mode.yaml"),
            "r") as fd:
        assert preset_def.get_environment_yaml() == fd.read()