TST: dt64 unit in tests (#56126)

Author: jbrockmendel

pandas/core/tools/datetimes.py

@@ -495,7 +495,8 @@ def _convert_listlike_datetimes(
  if tz_parsed is not None:
  # We can take a shortcut since the datetime64 numpy array
  # is in UTC
- dtype = cast(DatetimeTZDtype, tz_to_dtype(tz_parsed))
+ out_unit = np.datetime_data(result.dtype)[0]
+ dtype = cast(DatetimeTZDtype, tz_to_dtype(tz_parsed, out_unit))
  dt64_values = result.view(f"M8[{dtype.unit}]")
  dta = DatetimeArray._simple_new(dt64_values, dtype=dtype)
  return DatetimeIndex._simple_new(dta, name=name)

pandas/tests/arithmetic/test_timedelta64.py

@@ -1039,7 +1039,7 @@ def test_td64arr_add_datetime64_nat(self, box_with_array):
  other = np.datetime64("NaT")
 
  tdi = timedelta_range("1 day", periods=3)
- expected = DatetimeIndex(["NaT", "NaT", "NaT"])
+ expected = DatetimeIndex(["NaT", "NaT", "NaT"], dtype="M8[ns]")
 
  tdser = tm.box_expected(tdi, box_with_array)
  expected = tm.box_expected(expected, box_with_array)

pandas/tests/dtypes/test_dtypes.py

@@ -200,7 +200,8 @@ def test_is_boolean(self, categories, expected):
 
  def test_dtype_specific_categorical_dtype(self):
  expected = "datetime64[ns]"
- result = str(Categorical(DatetimeIndex([])).categories.dtype)
+ dti = DatetimeIndex([], dtype=expected)
+ result = str(Categorical(dti).categories.dtype)
  assert result == expected
 
  def test_not_string(self):

pandas/tests/frame/methods/test_transpose.py

@@ -29,7 +29,8 @@ def test_transpose_td64_intervals(self):
 
  def test_transpose_empty_preserves_datetimeindex(self):
  # GH#41382
- df = DataFrame(index=DatetimeIndex([]))
+ dti = DatetimeIndex([], dtype="M8[ns]")
+ df = DataFrame(index=dti)
 
  expected = DatetimeIndex([], dtype="datetime64[ns]", freq=None)
 

pandas/tests/indexes/datetimes/test_date_range.py

@@ -183,6 +183,7 @@ def test_date_range_edges(self, freq):
  )
  exp = DatetimeIndex(
  [ts + n * td for n in range(1, 5)],
+ dtype="M8[ns]",
  freq=freq,
  )
  tm.assert_index_equal(idx, exp)
@@ -193,7 +194,7 @@ def test_date_range_edges(self, freq):
  end=ts + td,
  freq=freq,
  )
- exp = DatetimeIndex([], freq=freq)
+ exp = DatetimeIndex([], dtype="M8[ns]", freq=freq)
  tm.assert_index_equal(idx, exp)
 
  # start matches end
@@ -202,7 +203,7 @@ def test_date_range_edges(self, freq):
  end=ts + td,
  freq=freq,
  )
- exp = DatetimeIndex([ts + td], freq=freq)
+ exp = DatetimeIndex([ts + td], dtype="M8[ns]", freq=freq)
  tm.assert_index_equal(idx, exp)
 
  def test_date_range_near_implementation_bound(self):

pandas/tests/indexes/datetimes/test_scalar_compat.py

@@ -304,7 +304,7 @@ def test_dti_fields(self, tz):
  exp = dti[[0, 90, 181, 273]]
  tm.assert_index_equal(res, exp)
  res = dti[dti.is_leap_year]
- exp = DatetimeIndex([], freq="D", tz=dti.tz, name="name")
+ exp = DatetimeIndex([], freq="D", tz=dti.tz, name="name").as_unit("ns")
  tm.assert_index_equal(res, exp)
 
  def test_dti_is_year_quarter_start(self):

pandas/tests/indexes/datetimes/test_setops.py

@@ -73,7 +73,7 @@ def test_union(self, tz, sort):
  expected2_notsorted = DatetimeIndex(list(other2) + list(rng2[:3]))
 
  rng3 = date_range("1/1/2000", freq="D", periods=5, tz=tz)
- other3 = DatetimeIndex([], tz=tz)
+ other3 = DatetimeIndex([], tz=tz).as_unit("ns")
  expected3 = date_range("1/1/2000", freq="D", periods=5, tz=tz)
  expected3_notsorted = rng3
 
@@ -235,7 +235,7 @@ def test_intersection(self, tz, sort):
  expected3 = date_range("6/1/2000", "6/20/2000", freq="D", name=None)
 
  rng4 = date_range("7/1/2000", "7/31/2000", freq="D", name="idx")
- expected4 = DatetimeIndex([], freq="D", name="idx")
+ expected4 = DatetimeIndex([], freq="D", name="idx", dtype="M8[ns]")
 
  for rng, expected in [
  (rng2, expected2),
@@ -265,7 +265,7 @@ def test_intersection(self, tz, sort):
 
  # GH 7880
  rng4 = date_range("7/1/2000", "7/31/2000", freq="D", tz=tz, name="idx")
- expected4 = DatetimeIndex([], tz=tz, name="idx")
+ expected4 = DatetimeIndex([], tz=tz, name="idx").as_unit("ns")
  assert expected4.freq is None
 
  for rng, expected in [
@@ -536,7 +536,7 @@ def test_intersection(self):
 
  # non-overlapping
  the_int = rng[:10].intersection(rng[10:])
- expected = DatetimeIndex([])
+ expected = DatetimeIndex([]).as_unit("ns")
  tm.assert_index_equal(the_int, expected)
 
  def test_intersection_bug(self):

pandas/tests/indexes/interval/test_interval.py

@@ -388,7 +388,7 @@ def test_maybe_convert_i8_nat(self, breaks):
  # GH 20636
  index = IntervalIndex.from_breaks(breaks)
 
- to_convert = breaks._constructor([pd.NaT] * 3)
+ to_convert = breaks._constructor([pd.NaT] * 3).as_unit("ns")
  expected = Index([np.nan] * 3, dtype=np.float64)
  result = index._maybe_convert_i8(to_convert)
  tm.assert_index_equal(result, expected)

pandas/tests/indexes/period/test_partial_slicing.py

@@ -117,7 +117,7 @@ def test_range_slice_outofbounds(self, make_range):
  idx = make_range(start="2013/10/01", freq="D", periods=10)
 
  df = DataFrame({"units": [100 + i for i in range(10)]}, index=idx)
- empty = DataFrame(index=type(idx)([], freq="D"), columns=["units"])
+ empty = DataFrame(index=idx[:0], columns=["units"])
  empty["units"] = empty["units"].astype("int64")
 
  tm.assert_frame_equal(df["2013/09/01":"2013/09/30"], empty)

pandas/tests/io/formats/test_format.py

@@ -872,20 +872,22 @@ def test_to_string_truncate_multilevel(self):
  with option_context("display.max_rows", 7, "display.max_columns", 7):
  assert has_doubly_truncated_repr(df)
 
- def test_truncate_with_different_dtypes(self):
+ @pytest.mark.parametrize("dtype", ["object", "datetime64[us]"])
+ def test_truncate_with_different_dtypes(self, dtype):
  # 11594, 12045
  # when truncated the dtypes of the splits can differ
 
  # 11594
- s = Series(
+ ser = Series(
  [datetime(2012, 1, 1)] * 10
  + [datetime(1012, 1, 2)]
- + [datetime(2012, 1, 3)] * 10
+ + [datetime(2012, 1, 3)] * 10,
+ dtype=dtype,
  )
 
  with option_context("display.max_rows", 8):
- result = str(s)
-  assert "object" in result
+ result = str(ser)
+ assert dtype in result
 
  def test_truncate_with_different_dtypes2(self):
  # 12045