Skip to content

Commit 5f4934f

Browse files
committed
TST: add index type coverage to timedelta subtraction tests (pandas-dev#31793)
1 parent 7d44431 commit 5f4934f

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

pandas/tests/arithmetic/test_timedelta64.py

+13-6
Original file line numberDiff line numberDiff line change
@@ -355,13 +355,15 @@ def test_subtraction_ops(self):
355355
expected = DatetimeIndex(["20121231", NaT, "20121230"], name="foo")
356356
tm.assert_index_equal(result, expected)
357357

358-
def test_subtraction_ops_with_tz(self):
358+
def test_subtraction_ops_with_tz(self, box_with_array):
359359

360360
# check that dt/dti subtraction ops with tz are validated
361361
dti = pd.date_range("20130101", periods=3)
362+
dti = tm.box_expected(dti, box_with_array)
362363
ts = Timestamp("20130101")
363364
dt = ts.to_pydatetime()
364365
dti_tz = pd.date_range("20130101", periods=3).tz_localize("US/Eastern")
366+
dti_tz = tm.box_expected(dti_tz, box_with_array)
365367
ts_tz = Timestamp("20130101").tz_localize("US/Eastern")
366368
ts_tz2 = Timestamp("20130101").tz_localize("CET")
367369
dt_tz = ts_tz.to_pydatetime()
@@ -412,27 +414,32 @@ def _check(result, expected):
412414

413415
result = dti_tz - dt_tz
414416
expected = TimedeltaIndex(["0 days", "1 days", "2 days"])
415-
tm.assert_index_equal(result, expected)
417+
expected = tm.box_expected(expected, box_with_array)
418+
tm.assert_equal(result, expected)
416419

417420
result = dt_tz - dti_tz
418421
expected = TimedeltaIndex(["0 days", "-1 days", "-2 days"])
419-
tm.assert_index_equal(result, expected)
422+
expected = tm.box_expected(expected, box_with_array)
423+
tm.assert_equal(result, expected)
420424

421425
result = dti_tz - ts_tz
422426
expected = TimedeltaIndex(["0 days", "1 days", "2 days"])
423-
tm.assert_index_equal(result, expected)
427+
expected = tm.box_expected(expected, box_with_array)
428+
tm.assert_equal(result, expected)
424429

425430
result = ts_tz - dti_tz
426431
expected = TimedeltaIndex(["0 days", "-1 days", "-2 days"])
427-
tm.assert_index_equal(result, expected)
432+
expected = tm.box_expected(expected, box_with_array)
433+
tm.assert_equal(result, expected)
428434

429435
result = td - td
430436
expected = Timedelta("0 days")
431437
_check(result, expected)
432438

433439
result = dti_tz - td
434440
expected = DatetimeIndex(["20121231", "20130101", "20130102"], tz="US/Eastern")
435-
tm.assert_index_equal(result, expected)
441+
expected = tm.box_expected(expected, box_with_array)
442+
tm.assert_equal(result, expected)
436443

437444
def test_dti_tdi_numeric_ops(self):
438445
# These are normally union/diff set-like ops

0 commit comments

Comments
 (0)