snapshot tests for plotting

Author: isabelizimm

setup.cfg

@@ -40,6 +40,7 @@ install_requires =
 dev =
     pytest
     pytest-cov
+    pytest-snapshot
     sphinx
     sphinx-autodoc-typehints
     sphinx-book-theme

vetiver/monitor.py

@@ -162,4 +162,5 @@ def plot_metrics(
     )
     fig.for_each_annotation(lambda a: a.update(text=a.text.split("=")[-1]))
     fig.update_layout(showlegend=False)
-    fig.show()
+
+    return fig

vetiver/tests/test_monitor.py

@@ -1,19 +1,23 @@
+from typing import List
 from sklearn import metrics
 from datetime import timedelta
 import pandas as pd
 import numpy
-
 import vetiver
 
 rng = pd.date_range("1/1/2012", periods=10, freq="S")
 new = dict(x=range(len(rng)), y=range(len(rng)))
 df = pd.DataFrame(new, index=rng)
-df.reset_index(inplace=True)
 td = timedelta(seconds=2)
 metric_set = [metrics.mean_squared_error, metrics.mean_absolute_error]
 
+def test_rolling():
+    m = [_ for _ in vetiver._rolling_df(df, td)]
+    assert len(m) == 5
+    assert len(m[0]) == 2
 
 def test_compute():
+    df.reset_index(inplace=True)
     m = vetiver.compute_metrics(
         df, "index", td, metric_set=metric_set, truth="x", estimate="y"
     )
@@ -23,3 +27,11 @@ def test_compute():
         m.metric.unique(),
         numpy.array(["mean_squared_error", "mean_absolute_error"], dtype=object),
     )
+
+def test_monitor(snapshot):
+    snapshot.snapshot_dir = './snapshots'
+    m = vetiver.compute_metrics(
+        df, "index", td, metric_set=metric_set, truth="x", estimate="y"
+    )
+    vetiver.plot_metrics(m)
+    snapshot.assert_match(m.to_json(), 'test_monitor.json')