Skip to content

Commit 75f2be6

Browse files
committed
TST: fix warnings in tests_timeseries_with_DateOffset
1 parent 12d9938 commit 75f2be6

File tree

1 file changed

+13
-12
lines changed

1 file changed

+13
-12
lines changed

pandas/tseries/tests/test_timeseries.py

+13-12
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from datetime import datetime, time, timedelta
44
import sys
55
import operator
6-
6+
import warnings
77
import nose
88

99
import numpy as np
@@ -2651,17 +2651,18 @@ def test_datetime64_with_DateOffset(self):
26512651
('WeekOfMonth', {'weekday': 2, 'week': 2}), 'Easter',
26522652
('DateOffset', {'day': 4}), ('DateOffset', {'month': 5})]
26532653

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)
26652666

26662667
# def test_add_timedelta64(self):
26672668
# rng = date_range('1/1/2000', periods=5)

0 commit comments

Comments
 (0)