diff --git a/tests/test_lumi_tools.py b/tests/test_lumi_tools.py index 9ad47a731..fb93ee8d7 100644 --- a/tests/test_lumi_tools.py +++ b/tests/test_lumi_tools.py @@ -191,6 +191,10 @@ def count_lumi(runs, lumis): total_lumi += my_lumidata.get_lumi(my_lumilist) return total_lumi + noclient_output = dask.compute(count_lumi(runs, lumis))[0] + with Client() as _: output = count_lumi(runs, lumis) - dask.compute(output)[0] + client_output = dask.compute(output)[0] + + assert noclient_output == client_output