tests for coerce datetime

Author: isabelizimm

vetiver/monitor.py

@@ -1,6 +1,6 @@
 import plotly.express as px
 import pandas as pd
-from datetime import timedelta
+from datetime import timedelta, datetime
 
 
 def compute_metrics(
@@ -42,7 +42,10 @@ def compute_metrics(
     """
 
     df = data[[truth, estimate, date_var]].copy()
-    df[date_var] = pd.to_datetime(df[date_var])
+
+    if not isinstance(date_var, datetime):
+        df[date_var] = pd.to_datetime(df[date_var])
+
     df = df.set_index(date_var).sort_index()
     lst = [_ for _ in _rolling_df(df=df, td=period)]
 

vetiver/tests/test_monitor.py

@@ -35,6 +35,27 @@ def test_compute():
     )
 
 
+def test_compute_coerce_datetime():
+
+    df_metrics = pd.DataFrame(
+        {
+            "index": ["2021-01-01", "2021-01-02", "2021-01-03"],
+            "truth": [200, 201, 199],
+            "pred": [198, 200, 199],
+        }
+    )
+    td = timedelta(days=1)
+    m = vetiver.compute_metrics(
+        df_metrics, "index", td, metric_set=metric_set, truth="truth", estimate="pred"
+    )
+    assert isinstance(m, pd.DataFrame)
+    assert m.shape == (4, 4)
+    numpy.testing.assert_array_equal(
+        m.metric.unique(),
+        numpy.array(["mean_squared_error", "mean_absolute_error"], dtype=object),
+    )
+
+
 def test_monitor(snapshot):
     snapshot.snapshot_dir = "./vetiver/tests/snapshots"
     m = vetiver.compute_metrics(