Refactor test_gil_contention

dask · Apr 26, 2023 · 9dc79c6 · 9dc79c6
1 parent c942dc9
commit 9dc79c6
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/distributed/tests/test_system_monitor.py b/distributed/tests/test_system_monitor.py
@@ -106,21 +106,17 @@ def test_gil_contention():
     a = sm.update()
     assert "gil_contention" in a
 
+    assert sm._gilknocker.is_running
+    sm.close()
+    sm.close()  # Idempotent
+    assert not sm._gilknocker.is_running
+
     sm = SystemMonitor(monitor_gil_contention=False)
     a = sm.update()
     assert "gil_contention" not in a
 
     assert dask.config.get("distributed.admin.system-monitor.gil.enabled")
-
     with dask.config.set({"distributed.admin.system-monitor.gil.enabled": False}):
         sm = SystemMonitor()
         a = sm.update()
         assert "gil_contention" not in a
-
-    sm = SystemMonitor()
-    a = sm.update()
-
-    assert sm._gilknocker.is_running
-    sm.close()
-    sm.close()  # Idempotent
-    assert not sm._gilknocker.is_running