added conftest to maek sure tests run using matplotlib

vertica · Jan 22, 2025 · c4eb8d7 · c4eb8d7
1 parent 7602452
commit c4eb8d7
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/verticapy/tests/vModel/conftest.py b/verticapy/tests/vModel/conftest.py
@@ -0,0 +1,17 @@
+# Pytest
+import pytest
+
+# VerticaPy
+import verticapy._config.config as conf
+
+# Other Modules
+
+
+@pytest.fixture(scope="module", autouse=True)
+def load_matplotlib():
+    """
+    Set default plotting library to matplotlib
+    """
+    conf.set_option("plotting_lib", "matplotlib")
+    yield
+    conf.set_option("plotting_lib", "plotly")