Merge pull request #370 from neutrinoceros/rfc/avoid_pyplot_in_tests_4

TST: avoid pyplot interface in tests (3/3)
volodia99 · Nov 2, 2024 · 8af9783 · 8af9783
2 parents a403cec + f5f158e
commit 8af9783
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,8 +1,8 @@
 from pathlib import Path
 
 import matplotlib
-import matplotlib.pyplot as plt
 import pytest
+from matplotlib.figure import Figure
 
 
 def pytest_configure(config):  # noqa: ARG001
@@ -26,6 +26,6 @@ def simulation_dir(test_data_dir, request):
 
 @pytest.fixture()
 def temp_figure_and_axis():
-    fig, ax = plt.subplots()
+    fig = Figure()
+    ax = fig.add_subplot()
     yield (fig, ax)
-    plt.close(fig)