ioos · ocefpaf · Jun 12, 2024 · Jun 6, 2024 · Jun 11, 2024 · Jun 11, 2024
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -36,4 +36,4 @@ jobs:
     - name: Full Tests
       shell: bash -l {0}
       run: |
-        python -m pytest -rxs --cov=gliderpy tests
+        python -m pytest -rxs --cov=gliderpy --mpl tests
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -12,6 +12,7 @@ pyarrow
 pytest
 pytest-cov
 pytest-flake8
+pytest-mpl
 pytest-sugar
 pytest-xdist
 setuptools_scm

diff --git a/tests/baseline/test_plot_transect.png b/tests/baseline/test_plot_transect.png
diff --git a/tests/test_plotting.py b/tests/test_plotting.py
@@ -0,0 +1,23 @@
+"""Test transect."""
+
+from pathlib import Path
+
+import pytest
+
+from gliderpy.fetchers import GliderDataFetcher
+from gliderpy.plotting import plot_transect
+
+root = Path(__file__).parent
+
+
-
+import matplotlib as mpl
+
+mpl.use("Agg")
-
+from pathlib import Path
+
+root = Path(__file__).parent
-
+import matplotlib as mpl
+
+mpl.use("Agg")
-
+from pathlib import Path
+
+root = Path(__file__).parent
+@pytest.mark.mpl_image_compare(baseline_dir=root.joinpath("baseline/"))
+def test_plot_transect():
+    glider_grab = GliderDataFetcher()
+
+    glider_grab.fetcher.dataset_id = "whoi_406-20160902T1700"
+    df = glider_grab.to_pandas()
+    # Generate the plot
+    fig, ax = plot_transect(df, "temperature")
+
+    # Return the figure for pytest-mpl to compare
+    return fig