@@ -368,20 +368,25 @@ def test_dti_add_offset_array(self, tz, box):
368
368
# GH#18849
369
369
dti = pd .date_range ('2017-01-01' , periods = 2 , tz = tz )
370
370
other = box ([pd .offsets .MonthEnd (), pd .offsets .Day (n = 2 )])
371
- res = dti + other
371
+
372
+ with tm .assert_produces_warning (PerformanceWarning ):
373
+ res = dti + other
372
374
expected = DatetimeIndex ([dti [n ] + other [n ] for n in range (len (dti ))],
373
375
name = dti .name , freq = 'infer' )
374
376
tm .assert_index_equal (res , expected )
375
377
376
- res2 = other + dti
378
+ with tm .assert_produces_warning (PerformanceWarning ):
379
+ res2 = other + dti
377
380
tm .assert_index_equal (res2 , expected )
378
381
379
382
@pytest .mark .parametrize ('box' , [np .array , pd .Index ])
380
383
def test_dti_sub_offset_array (self , tz , box ):
381
384
# GH#18824
382
385
dti = pd .date_range ('2017-01-01' , periods = 2 , tz = tz )
383
386
other = box ([pd .offsets .MonthEnd (), pd .offsets .Day (n = 2 )])
384
- res = dti - other
387
+
388
+ with tm .assert_produces_warning (PerformanceWarning ):
389
+ res = dti - other
385
390
expected = DatetimeIndex ([dti [n ] - other [n ] for n in range (len (dti ))],
386
391
name = dti .name , freq = 'infer' )
387
392
tm .assert_index_equal (res , expected )
@@ -392,20 +397,25 @@ def test_dti_sub_offset_array(self, tz, box):
392
397
def test_dti_with_offset_series (self , tz , names ):
393
398
# GH#18849
394
399
dti = pd .date_range ('2017-01-01' , periods = 2 , tz = tz , name = names [0 ])
395
- other = pd . Series ([pd .offsets .MonthEnd (), pd .offsets .Day (n = 2 )],
396
- name = names [1 ])
400
+ other = Series ([pd .offsets .MonthEnd (), pd .offsets .Day (n = 2 )],
401
+ name = names [1 ])
397
402
398
- expected_add = pd .Series ([dti [n ] + other [n ] for n in range (len (dti ))],
399
- name = names [2 ])
400
- res = dti + other
403
+ expected_add = Series ([dti [n ] + other [n ] for n in range (len (dti ))],
404
+ name = names [2 ])
405
+
406
+ with tm .assert_produces_warning (PerformanceWarning ):
407
+ res = dti + other
401
408
tm .assert_series_equal (res , expected_add )
402
- res2 = other + dti
409
+
410
+ with tm .assert_produces_warning (PerformanceWarning ):
411
+ res2 = other + dti
403
412
tm .assert_series_equal (res2 , expected_add )
404
413
405
- expected_sub = pd . Series ([dti [n ] - other [n ] for n in range (len (dti ))],
406
- name = names [2 ])
414
+ expected_sub = Series ([dti [n ] - other [n ] for n in range (len (dti ))],
415
+ name = names [2 ])
407
416
408
- res3 = dti - other
417
+ with tm .assert_produces_warning (PerformanceWarning ):
418
+ res3 = dti - other
409
419
tm .assert_series_equal (res3 , expected_sub )
410
420
411
421
0 commit comments