|
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