TST: improve state cleanup in test fixture

volodia99 · Nov 2, 2024 · 13d9bcb · 13d9bcb
1 parent 3002b83
commit 13d9bcb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_image_comp.py b/tests/test_image_comp.py
@@ -1,3 +1,5 @@
+from copy import deepcopy
+
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 import numpy as np
@@ -11,8 +13,10 @@
 @pytest.fixture
 def tmp_mpl_state():
     # reset matplotlib's state when the test is over
-    style = mpl.rcParams.copy()
+    style = deepcopy(mpl.rcParams)
     yield
+    for k in set(mpl.rcParams.keys()) - set(style.keys()):
+        mpl.rcParams.pop(k)
     mpl.rcParams.update(style)