|
3 | 3 | from datetime import datetime, time, timedelta
|
4 | 4 | import sys
|
5 | 5 | import operator
|
6 |
| - |
| 6 | +import warnings |
7 | 7 | import nose
|
8 | 8 |
|
9 | 9 | import numpy as np
|
@@ -2651,17 +2651,18 @@ def test_datetime64_with_DateOffset(self):
|
2651 | 2651 | ('WeekOfMonth', {'weekday': 2, 'week': 2}), 'Easter',
|
2652 | 2652 | ('DateOffset', {'day': 4}), ('DateOffset', {'month': 5})]
|
2653 | 2653 |
|
2654 |
| - for normalize in (True, False): |
2655 |
| - for do in offsets: |
2656 |
| - if isinstance(do, tuple): |
2657 |
| - do, kwargs = do |
2658 |
| - else: |
2659 |
| - do = do |
2660 |
| - kwargs = {} |
2661 |
| - op = getattr(pd.offsets,do)(5, normalize=normalize, **kwargs) |
2662 |
| - assert_func(klass([x + op for x in s]), s + op) |
2663 |
| - assert_func(klass([x - op for x in s]), s - op) |
2664 |
| - assert_func(klass([op + x for x in s]), op + s) |
| 2654 | + with warnings.catch_warnings(record=True): |
| 2655 | + for normalize in (True, False): |
| 2656 | + for do in offsets: |
| 2657 | + if isinstance(do, tuple): |
| 2658 | + do, kwargs = do |
| 2659 | + else: |
| 2660 | + do = do |
| 2661 | + kwargs = {} |
| 2662 | + op = getattr(pd.offsets,do)(5, normalize=normalize, **kwargs) |
| 2663 | + assert_func(klass([x + op for x in s]), s + op) |
| 2664 | + assert_func(klass([x - op for x in s]), s - op) |
| 2665 | + assert_func(klass([op + x for x in s]), op + s) |
2665 | 2666 |
|
2666 | 2667 | # def test_add_timedelta64(self):
|
2667 | 2668 | # rng = date_range('1/1/2000', periods=5)
|
|
0 commit comments