From 86c2a33a02b46c126bba87edc4c478f3cfcf5c31 Mon Sep 17 00:00:00 2001
From: Oksana Shadura <ksu.shadura@gmail.com>
Date: Tue, 13 Feb 2024 14:45:36 +0100
Subject: [PATCH] Switch test dimuon for coffeav0 to test Dask executor

---
 tests/test_dimuon.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/test_dimuon.py b/tests/test_dimuon.py
index 3ace646..97b7552 100644
--- a/tests/test_dimuon.py
+++ b/tests/test_dimuon.py
@@ -5,6 +5,8 @@
 from coffea.nanoevents.methods import candidate
 from coffea.nanoevents import NanoEventsFactory, BaseSchema
 
+from dask.distributed import Client
+
 fileset = {
     'DoubleMuon': [
         'root://eospublic.cern.ch//eos/root-eos/cms_opendata_2012_nanoaod/Run2012B_DoubleMuParked.root',
@@ -59,11 +61,12 @@ def process(self, events):
     def postprocess(self, accumulator):
         pass
 
-#@pytest.mark.coffeav0
+@pytest.mark.coffeav0
 def test_processor_dimu_mass():
-    iterative_run = processor.Runner(executor = processor.IterativeExecutor(compression=None),
+    client = Client()
+    iterative_run = processor.Runner(executor = processor.dask_executor,
                                      schema=BaseSchema,
-                                     maxchunks=10
+                                     client=client,
                                      )
     out = iterative_run(fileset,
                         treename="Events",