Python parse_step_selectionの例

コード例 #1

def test_parse_step_selection_invalid():

    with pytest.raises(
            DagsterInvalidSubsetError,
            match="No qualified steps to execute found for step_selection",
    ):
        parse_step_selection(step_deps, ["1+some_solid"])

コード例 #2

ファイル: test_subset_selector.py プロジェクト: sd2k/dagster

def test_parse_step_selection_multi():
    step_selection_multi_disjoint = parse_step_selection(
        step_deps, ["return_one.compute", "add_nums.compute+"]
    )
    assert len(step_selection_multi_disjoint) == 3
    assert set(step_selection_multi_disjoint) == {
        "return_one.compute",
        "add_nums.compute",
        "multiply_two.compute",
    }

    step_selection_multi_overlap = parse_step_selection(
        step_deps, ["*add_nums.compute", "return_one.compute+"]
    )
    assert len(step_selection_multi_overlap) == 3
    assert set(step_selection_multi_overlap) == {
        "return_one.compute",
        "return_two.compute",
        "add_nums.compute",
    }

    with pytest.raises(
        DagsterInvalidSubsetError, match="No qualified steps to execute found for step_selection",
    ):
        parse_step_selection(step_deps, ["*add_nums.compute", "a"])

コード例 #3

def test_parse_step_selection_single():
    step_selection_single = parse_step_selection(step_deps, ["add_nums"])
    assert len(step_selection_single) == 1
    assert step_selection_single == {"add_nums"}

    step_selection_star = parse_step_selection(step_deps, ["add_nums*"])
    assert len(step_selection_star) == 3
    assert set(step_selection_star) == {
        "add_nums",
        "multiply_two",
        "add_one",
    }

    step_selection_both = parse_step_selection(step_deps, ["*add_nums+"])
    assert len(step_selection_both) == 4
    assert set(step_selection_both) == {
        "return_one",
        "return_two",
        "add_nums",
        "multiply_two",
    }

コード例 #4

def test_parse_step_selection_multi():
    step_selection_multi_disjoint = parse_step_selection(
        step_deps, ["return_one", "add_nums+"])
    assert len(step_selection_multi_disjoint) == 3
    assert set(step_selection_multi_disjoint) == {
        "return_one",
        "add_nums",
        "multiply_two",
    }

    step_selection_multi_overlap = parse_step_selection(
        step_deps, ["*add_nums", "return_one+"])
    assert len(step_selection_multi_overlap) == 3
    assert set(step_selection_multi_overlap) == {
        "return_one",
        "return_two",
        "add_nums",
    }

    with pytest.raises(
            DagsterExecutionStepNotFoundError,
            match="Step selection refers to unknown step: a",
    ):
        parse_step_selection(step_deps, ["*add_nums", "a"])