Force synchronous (serial) computation in tests

datasciencecampus · Nov 28, 2023 · 02496f6 · 02496f6
1 parent 034fdca
commit 02496f6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/tests/mst/test_measure.py b/tests/mst/test_measure.py
@@ -3,6 +3,7 @@
 import platform
 from unittest import mock
 
+import dask
 import numpy as np
 import pandas as pd
 import pytest
@@ -63,7 +64,9 @@ def test_measure(params, num_cliques):
     population_type, area_type, dimensions, clique, table = params
     mst = mocked_mst(population_type, area_type, dimensions)
 
-    with mock.patch("centhesus.mst.MST.get_marginal") as get_marginal:
+    with mock.patch(
+        "centhesus.mst.MST.get_marginal"
+    ) as get_marginal, dask.config.set(scheduler="synchronous"):
         get_marginal.return_value = table
         measurements = mst.measure([clique] * num_cliques)
 

diff --git a/tests/mst/test_select.py b/tests/mst/test_select.py
@@ -4,6 +4,7 @@
 import platform
 from unittest import mock
 
+import dask
 import networkx as nx
 import pytest
 from hypothesis import given, settings
@@ -71,7 +72,9 @@ def test_calculate_importances(params):
     population_type, area_type, dimensions, domain, importances = params
     mst = mocked_mst(population_type, area_type, dimensions, domain=domain)
 
-    with mock.patch("centhesus.mst.MST._calculate_importance_of_pair") as calc:
+    with mock.patch(
+        "centhesus.mst.MST._calculate_importance_of_pair"
+    ) as calc, dask.config.set(scheduler="synchronous"):
         calc.side_effect = importances
         weights = mst._calculate_importances("interim")