fix tests

Author: AgnieszkaZaba

examples/notebook_vars.ipynb

@@ -12,8 +12,8 @@
     "from matplotlib import pyplot \n",
     "from open_atmos_jupyter_utils import show_plot\n",
     "\n",
-    "pyplot.plot([])\n",
-    "show_plot()"
+    "fig = pyplot.figure()\n",
+    "show_plot(fig)"
    ],
    "id": "840cd46405eb9d0e",
    "outputs": [

tests/test_notebook_vars.py

@@ -6,14 +6,14 @@
 import pytest
 import matplotlib
 
+
 from open_atmos_jupyter_utils import notebook_vars
-import examples
 
 @pytest.fixture(scope="session", name="notebook_variables")
 def notebook_variables_fixture():
     """returns variables from the notebook """
     return notebook_vars(
-        file=Path(examples.__file__).parent / "notebook_vars.ipynb",
+        file=str(Path(__file__).parent.parent / "examples" / "notebook_vars.ipynb"),
         plot=False,
     )
 class TestNotebookVars:
@@ -24,6 +24,7 @@ def test_notebook_vars(notebook_variables):
         assert notebook_variables["c"] == notebook_variables["a"] + notebook_variables["b"]
 
     @staticmethod
-    def test_plots_closed():
+    def test_plots_closed(notebook_variables):
         """ checks all figures closed """
+        assert isinstance(notebook_variables["fig"], matplotlib.figure.Figure) 
         assert 0 == len(matplotlib.pyplot.get_fignums())