From ecd1741e002d829ade5fb7c9f51e956adaeb6c08 Mon Sep 17 00:00:00 2001 From: Lindsey Gray Date: Wed, 29 Jan 2025 10:45:57 -0600 Subject: [PATCH] harden test a little --- tests/test_lumi_tools.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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