Merge pull request #107 from FloraSauerbronn/create-test-transect

Adding test_plotting in tests

.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

requirements-dev.txt

@@ -12,6 +12,7 @@ pyarrow
 pytest
 pytest-cov
 pytest-flake8
+pytest-mpl
 pytest-sugar
 pytest-xdist
 setuptools_scm

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
+
+
+@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