From 05bbd5954bd728353d3c924232ebc4eb69675955 Mon Sep 17 00:00:00 2001 From: Joe Date: Thu, 21 Dec 2023 09:25:30 -0600 Subject: [PATCH] fix: Fixed time preprocessor test (#181) Fixed time preprocessor tests that were failing due to more granular engine types --- .../plot/express/preprocess/test_TimePreprocessor.py | 5 +++-- 1 file changed, 3 insertions(+), 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 index c19629815..29e131bfa 100644 --- 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)