Skip to content

Commit 3c89a50

Browse files
committed
TST: add index type coverage to timedelta subtraction tests (pandas-dev#31793)
1 parent fbbac5e commit 3c89a50

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()
@@ -411,27 +413,32 @@ def _check(result, expected):
411413

412414
result = dti_tz - dt_tz
413415
expected = TimedeltaIndex(["0 days", "1 days", "2 days"])
414-
tm.assert_index_equal(result, expected)
416+
expected = tm.box_expected(expected, box_with_array)
417+
tm.assert_equal(result, expected)
415418

416419
result = dt_tz - dti_tz
417420
expected = TimedeltaIndex(["0 days", "-1 days", "-2 days"])
418-
tm.assert_index_equal(result, expected)
421+
expected = tm.box_expected(expected, box_with_array)
422+
tm.assert_equal(result, expected)
419423

420424
result = dti_tz - ts_tz
421425
expected = TimedeltaIndex(["0 days", "1 days", "2 days"])
422-
tm.assert_index_equal(result, expected)
426+
expected = tm.box_expected(expected, box_with_array)
427+
tm.assert_equal(result, expected)
423428

424429
result = ts_tz - dti_tz
425430
expected = TimedeltaIndex(["0 days", "-1 days", "-2 days"])
426-
tm.assert_index_equal(result, expected)
431+
expected = tm.box_expected(expected, box_with_array)
432+
tm.assert_equal(result, expected)
427433

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

432438
result = dti_tz - td
433439
expected = DatetimeIndex(["20121231", "20130101", "20130102"], tz="US/Eastern")
434-
tm.assert_index_equal(result, expected)
440+
expected = tm.box_expected(expected, box_with_array)
441+
tm.assert_equal(result, expected)
435442

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

0 commit comments

Comments
 (0)