parametrize over unit

Author: jbrockmendel

pandas/tests/reshape/test_cut.py

@@ -592,18 +592,18 @@ def test_datetime_nan_mask():
 
 
 @pytest.mark.parametrize("tz", [None, "UTC", "US/Pacific"])
-def test_datetime_cut_roundtrip(tz):
+def test_datetime_cut_roundtrip(tz, unit):
     # see gh-19891
-    ser = Series(date_range("20180101", periods=3, tz=tz))
+    ser = Series(date_range("20180101", periods=3, tz=tz, unit=unit))
     result, result_bins = cut(ser, 2, retbins=True)
 
     expected = cut(ser, result_bins)
     tm.assert_series_equal(result, expected)
 
     expected_bins = DatetimeIndex(
-        ["2017-12-31 23:57:07.200000", "2018-01-02 00:00:00", "2018-01-03 00:00:00"]
-    ).as_unit("ns")
-    expected_bins = expected_bins.tz_localize(tz)
+        ["2017-12-31 23:57:07.200000", "2018-01-02 00:00:00", "2018-01-03 00:00:00"],
+        dtype=f"M8[{unit}]",
+    ).tz_localize(tz)
     tm.assert_index_equal(result_bins, expected_bins)
 
 

pandas/tests/reshape/test_qcut.py

@@ -268,8 +268,10 @@ def test_datetime_tz_qcut(bins):
         ],
     ],
 )
-def test_date_like_qcut_bins(arg, expected_bins):
+def test_date_like_qcut_bins(arg, expected_bins, unit):
     # see gh-19891
+    arg = arg.as_unit(unit)
+    expected_bins = expected_bins.as_unit(unit)
     ser = Series(arg)
     result, result_bins = qcut(ser, 2, retbins=True)
     tm.assert_index_equal(result_bins, expected_bins)