Exemplos de define_multi_mode_pipeline em Python

Exemplo n.º 1

def test_execute_multi_mode_errors():
    multi_mode_pipeline = define_multi_mode_pipeline()

    with pytest.raises(DagsterInvariantViolationError):
        execute_pipeline(multi_mode_pipeline)

    with pytest.raises(DagsterInvariantViolationError):
        execute_pipeline(pipeline=multi_mode_pipeline, mode="wrong_mode")

Exemplo n.º 2

def test_execute_multi_mode():
    multi_mode_pipeline = define_multi_mode_pipeline()

    assert (execute_pipeline(
        pipeline=multi_mode_pipeline,
        mode="mode_one").result_for_solid("return_three").output_value() == 3)

    assert (execute_pipeline(
        pipeline=multi_mode_pipeline,
        mode="mode_two").result_for_solid("return_three").output_value() == 3)