Figurewidget datetime bug (#232)

* 💨 adding tests for #231

* 🙈 formatting

* 🙏 bugfix

* 💪 fixing 🐛

Author: jonasvdd

plotly_resampler/figure_resampler/figure_resampler_interface.py

@@ -600,7 +600,9 @@ def _parse_get_trace_props(
             # If we cast a tz-aware datetime64 array to `.values` we lose the tz-info 
             # and the UTC time will be displayed instead of the tz-localized time, 
             # hence we cast to a pd.DatetimeIndex, which preserves the tz-info
-            else pd.Index(hf_x) if pd.core.dtypes.common.is_datetime64tz_dtype(hf_x)
+            # As a matter of fact, to resolve #231, we also convert non-tz-aware 
+            # datetime64 arrays to an pd.Index
+            else pd.Index(hf_x) if pd.core.dtypes.common.is_datetime64_any_dtype(hf_x)
             else hf_x.values if isinstance(hf_x, pd.Series)
             else hf_x if isinstance(hf_x, pd.Index)
             else np.asarray(hf_x)

tests/test_figure_resampler.py

@@ -786,6 +786,7 @@ def test_datetime_hf_x_no_index():
         }
     )
     assert len(output) == 2
+    assert isinstance(output[1]["x"], pd.Index)
 
     # add via scatter kwargs
     fr = FigureResampler()
@@ -797,6 +798,7 @@ def test_datetime_hf_x_no_index():
         }
     )
     assert len(output) == 2
+    assert isinstance(output[1]["x"], pd.Index)
 
 
 def test_multiple_timezones_in_single_x_index__datetimes_and_timestamps():

tests/test_figurewidget_resampler.py

@@ -548,6 +548,7 @@ def test_datetime_hf_x_no_index():
         }
     )
     assert len(output) == 2
+    assert isinstance(output[1]["x"], pd.Index)
 
     # add via scatter kwargs
     fr = FigureWidgetResampler()
@@ -559,6 +560,7 @@ def test_datetime_hf_x_no_index():
         }
     )
     assert len(output) == 2
+    assert isinstance(output[1]["x"], pd.Index)
 
 
 def test_multiple_timezones_in_single_x_index__datetimes_and_timestamps():