Function test_array_compute

dask/diagnostics/tests/test_progress.py:24–32

(capsys)

def test_array_compute(capsys):
    pytest.importorskip("numpy")
    da = pytest.importorskip("dask.array")

    data = da.ones((100, 100), dtype="f4", chunks=(100, 100))
    with ProgressBar():
        out = data.sum().compute()
    assert out == 10000
    check_bar_completed(capsys)


def test_progressbar(capsys):