TST: dont skip dt64tz rolling tests (#44730)

Author: jbrockmendel

pandas/tests/window/conftest.py

@@ -200,13 +200,7 @@ def halflife_with_times(request):
  "float64",
  "m8[ns]",
  "M8[ns]",
- pytest.param(
- "datetime64[ns, UTC]",
- marks=pytest.mark.skip(
- "direct creation of extension dtype datetime64[ns, UTC] "
- "is not supported ATM"
- ),
- ),
+ "datetime64[ns, UTC]",
  ]
 )
 def dtypes(request):

pandas/tests/window/test_dtypes.py

@@ -1,6 +1,8 @@
 import numpy as np
 import pytest
 
+from pandas.core.dtypes.common import pandas_dtype
+
 from pandas import (
  NA,
  DataFrame,
@@ -19,9 +21,7 @@
 def get_dtype(dtype, coerce_int=None):
  if coerce_int is False and "int" in dtype:
  return None
- if dtype != "category":
- return np.dtype(dtype)
- return dtype
+ return pandas_dtype(dtype)
 
 
 @pytest.mark.parametrize(
@@ -66,21 +66,23 @@ def get_dtype(dtype, coerce_int=None):
  ],
 )
 def test_series_dtypes(method, data, expected_data, coerce_int, dtypes, min_periods):
- s = Series(data, dtype=get_dtype(dtypes, coerce_int=coerce_int))
- if dtypes in ("m8[ns]", "M8[ns]") and method != "count":
+ ser = Series(data, dtype=get_dtype(dtypes, coerce_int=coerce_int))
+ rolled = ser.rolling(2, min_periods=min_periods)
+
+ if dtypes in ("m8[ns]", "M8[ns]", "datetime64[ns, UTC]") and method != "count":
  msg = "No numeric types to aggregate"
  with pytest.raises(DataError, match=msg):
- getattr(s.rolling(2, min_periods=min_periods), method)()
+ getattr(rolled, method)()
  else:
- result = getattr(s.rolling(2, min_periods=min_periods), method)()
+ result = getattr(rolled, method)()
  expected = Series(expected_data, dtype="float64")
  tm.assert_almost_equal(result, expected)
 
 
 def test_series_nullable_int(any_signed_int_ea_dtype):
  # GH 43016
- s = Series([0, 1, NA], dtype=any_signed_int_ea_dtype)
- result = s.rolling(2).mean()
+ ser = Series([0, 1, NA], dtype=any_signed_int_ea_dtype)
+ result = ser.rolling(2).mean()
  expected = Series([np.nan, 0.5, np.nan])
  tm.assert_series_equal(result, expected)
 
@@ -130,14 +132,15 @@ def test_series_nullable_int(any_signed_int_ea_dtype):
  ],
 )
 def test_dataframe_dtypes(method, expected_data, dtypes, min_periods):
- if dtypes == "category":
- pytest.skip("Category dataframe testing not implemented.")
+
  df = DataFrame(np.arange(10).reshape((5, 2)), dtype=get_dtype(dtypes))
- if dtypes in ("m8[ns]", "M8[ns]") and method != "count":
+ rolled = df.rolling(2, min_periods=min_periods)
+
+ if dtypes in ("m8[ns]", "M8[ns]", "datetime64[ns, UTC]") and method != "count":
  msg = "No numeric types to aggregate"
  with pytest.raises(DataError, match=msg):
- getattr(df.rolling(2, min_periods=min_periods), method)()
+ getattr(rolled, method)()
  else:
- result = getattr(df.rolling(2, min_periods=min_periods), method)()
+ result = getattr(rolled, method)()
  expected = DataFrame(expected_data, dtype="float64")
  tm.assert_frame_equal(result, expected)