fix: Fixed time preprocessor test (#181)

Fixed time preprocessor tests that were failing due to more granular engine types
deephaven · Dec 21, 2023 · 05bbd59 · 05bbd59
1 parent cedcd3c
commit 05bbd59
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/plugins/plotly-express/test/deephaven/plot/express/preprocess/test_TimePreprocessor.py b/plugins/plotly-express/test/deephaven/plot/express/preprocess/test_TimePreprocessor.py
@@ -40,14 +40,15 @@ def test_time_preprocessor(self):
 
         expected_df = pd.DataFrame(
             {
-                "Start": ["2021-07-04 12:00:00"],
-                "End": ["2021-07-04 13:00:00"],
+                "Start": ["2021-07-04 12:00:00+00:00"],
+                "End": ["2021-07-04 13:00:00+00:00"],
                 "Category": ["A"],
                 "x_diff": [3600000.0],
             }
         )
         expected_df["Start"] = pd.to_datetime(expected_df["Start"])
         expected_df["End"] = pd.to_datetime(expected_df["End"])
+        expected_df["Category"] = expected_df["Category"].astype("string")
         expected_df["x_diff"] = expected_df["x_diff"].astype("Float64")
 
         new_df = dhpd.to_pandas(new_table)