Пример #1
0
def test_execute_summary_stats_error():
    with pytest.raises(DagsterInvariantViolationError):
        assert _execute_summary_stats('foo', DataFrame({}), lambda value: 'jajaja')

    with pytest.raises(DagsterInvariantViolationError):
        assert _execute_summary_stats(
            'foo',
            DataFrame({}),
            lambda value: [EventMetadataEntry.text('baz', 'qux', 'quux'), 'rofl'],
        )
Пример #2
0
def test_execute_summary_stats_error():
    with pytest.raises(DagsterInvariantViolationError):
        assert _execute_summary_stats("foo", DataFrame({}), lambda value: "jajaja")

    with pytest.raises(DagsterInvariantViolationError):
        assert _execute_summary_stats(
            "foo",
            DataFrame({}),
            lambda value: [EventMetadataEntry.text("baz", "qux", "quux"), "rofl"],
        )
Пример #3
0
def test_execute_summary_stats_null_function():
    assert _execute_summary_stats("foo", DataFrame(), None) == []

    metadata_entries = _execute_summary_stats(
        "foo",
        DataFrame({"bar": [1, 2, 3]}),
        lambda value: [MetadataEntry("qux", value="baz")],
    )
    assert len(metadata_entries) == 1
    assert metadata_entries[0].label == "qux"
    assert metadata_entries[0].entry_data.text == "baz"
Пример #4
0
def test_execute_summary_stats_null_function():
    assert _execute_summary_stats('foo', DataFrame(), None) == []

    metadata_entries = _execute_summary_stats(
        'foo',
        DataFrame({'bar': [1, 2, 3]}),
        lambda value: [EventMetadataEntry.text('baz', 'qux', 'quux')],
    )
    assert len(metadata_entries) == 1
    assert metadata_entries[0].label == 'qux'
    assert metadata_entries[0].description == 'quux'
    assert metadata_entries[0].entry_data.text == 'baz'
Пример #5
0
def test_execute_summary_stats_metadata_value_error():
    with pytest.raises(DagsterInvariantViolationError):
        assert _execute_summary_stats(
            "foo", DataFrame({}), event_metadata_fn=lambda _: {"bad": None})