Exemplos de define_single_mode_pipeline em Python

Exemplo n.º 1

Arquivo: test_mode_definitions.py Projeto: sd2k/dagster

def test_wrong_single_mode():
    with pytest.raises(DagsterInvariantViolationError):
        assert (
            execute_pipeline(pipeline=define_single_mode_pipeline(), mode="wrong_mode")
            .result_for_solid("return_two")
            .output_value()
            == 2
        )

Exemplo n.º 2

def test_execute_single_mode():
    single_mode_pipeline = define_single_mode_pipeline()
    assert single_mode_pipeline.is_single_mode is True

    assert execute_pipeline(single_mode_pipeline).result_for_solid(
        "return_two").output_value() == 2

    assert (execute_pipeline(
        single_mode_pipeline,
        mode="the_mode").result_for_solid("return_two").output_value() == 2)