6
6
PeriodIndex , TimedeltaIndex , Timedelta , timedelta_range ,
7
7
date_range , Float64Index )
8
8
import pandas .tslib as tslib
9
+ import pandas .tseries .period as period
9
10
10
11
import pandas .util .testing as tm
11
12
@@ -1617,9 +1618,9 @@ def test_add_iadd(self):
1617
1618
for o in [pd .offsets .YearBegin (2 ), pd .offsets .MonthBegin (1 ),
1618
1619
pd .offsets .Minute (), np .timedelta64 (365 , 'D' ),
1619
1620
timedelta (365 ), Timedelta (days = 365 )]:
1620
- msg = 'Input has different freq from PeriodIndex \\ (freq=A-DEC \\ ) '
1621
- with tm . assertRaisesRegexp ( ValueError ,
1622
- 'Input has different freq from Period' ):
1621
+ msg = ( 'Input has different freq(=.+)? '
1622
+ 'from PeriodIndex \\ (freq=A-DEC \\ )' )
1623
+ with tm . assertRaisesRegexp ( period . IncompatibleFrequency , msg ):
1623
1624
rng + o
1624
1625
1625
1626
rng = pd .period_range ('2014-01' , '2016-12' , freq = 'M' )
@@ -1633,8 +1634,8 @@ def test_add_iadd(self):
1633
1634
pd .offsets .Minute (), np .timedelta64 (365 , 'D' ),
1634
1635
timedelta (365 ), Timedelta (days = 365 )]:
1635
1636
rng = pd .period_range ('2014-01' , '2016-12' , freq = 'M' )
1636
- msg = 'Input has different freq from PeriodIndex\\ (freq=M\\ )'
1637
- with tm .assertRaisesRegexp (ValueError , msg ):
1637
+ msg = 'Input has different freq(=.+)? from PeriodIndex\\ (freq=M\\ )'
1638
+ with tm .assertRaisesRegexp (period . IncompatibleFrequency , msg ):
1638
1639
rng + o
1639
1640
1640
1641
# Tick
@@ -1654,8 +1655,8 @@ def test_add_iadd(self):
1654
1655
pd .offsets .Minute (), np .timedelta64 (4 , 'h' ),
1655
1656
timedelta (hours = 23 ), Timedelta ('23:00:00' )]:
1656
1657
rng = pd .period_range ('2014-05-01' , '2014-05-15' , freq = 'D' )
1657
- msg = 'Input has different freq from PeriodIndex\\ (freq=D\\ )'
1658
- with tm .assertRaisesRegexp (ValueError , msg ):
1658
+ msg = 'Input has different freq(=.+)? from PeriodIndex\\ (freq=D\\ )'
1659
+ with tm .assertRaisesRegexp (period . IncompatibleFrequency , msg ):
1659
1660
rng + o
1660
1661
1661
1662
offsets = [pd .offsets .Hour (2 ), timedelta (hours = 2 ),
@@ -1676,10 +1677,10 @@ def test_add_iadd(self):
1676
1677
np .timedelta64 (30 , 's' ), Timedelta (seconds = 30 )]:
1677
1678
rng = pd .period_range ('2014-01-01 10:00' , '2014-01-05 10:00' ,
1678
1679
freq = 'H' )
1679
- msg = 'Input has different freq from PeriodIndex\\ (freq=H\\ )'
1680
- with tm .assertRaisesRegexp (ValueError , msg ):
1680
+ msg = 'Input has different freq(=.+)? from PeriodIndex\\ (freq=H\\ )'
1681
+ with tm .assertRaisesRegexp (period . IncompatibleFrequency , msg ):
1681
1682
result = rng + delta
1682
- with tm .assertRaisesRegexp (ValueError , msg ):
1683
+ with tm .assertRaisesRegexp (period . IncompatibleFrequency , msg ):
1683
1684
rng += delta
1684
1685
1685
1686
# int
@@ -1745,8 +1746,9 @@ def test_sub_isub(self):
1745
1746
pd .offsets .Minute (), np .timedelta64 (365 , 'D' ),
1746
1747
timedelta (365 )]:
1747
1748
rng = pd .period_range ('2014' , '2024' , freq = 'A' )
1748
- msg = 'Input has different freq from PeriodIndex\\ (freq=A-DEC\\ )'
1749
- with tm .assertRaisesRegexp (ValueError , msg ):
1749
+ msg = ('Input has different freq(=.+)? '
1750
+ 'from PeriodIndex\\ (freq=A-DEC\\ )' )
1751
+ with tm .assertRaisesRegexp (period .IncompatibleFrequency , msg ):
1750
1752
rng - o
1751
1753
1752
1754
rng = pd .period_range ('2014-01' , '2016-12' , freq = 'M' )
@@ -1760,8 +1762,8 @@ def test_sub_isub(self):
1760
1762
pd .offsets .Minute (), np .timedelta64 (365 , 'D' ),
1761
1763
timedelta (365 )]:
1762
1764
rng = pd .period_range ('2014-01' , '2016-12' , freq = 'M' )
1763
- msg = 'Input has different freq from PeriodIndex\\ (freq=M\\ )'
1764
- with tm .assertRaisesRegexp (ValueError , msg ):
1765
+ msg = 'Input has different freq(=.+)? from PeriodIndex\\ (freq=M\\ )'
1766
+ with tm .assertRaisesRegexp (period . IncompatibleFrequency , msg ):
1765
1767
rng - o
1766
1768
1767
1769
# Tick
@@ -1780,8 +1782,8 @@ def test_sub_isub(self):
1780
1782
pd .offsets .Minute (), np .timedelta64 (4 , 'h' ),
1781
1783
timedelta (hours = 23 )]:
1782
1784
rng = pd .period_range ('2014-05-01' , '2014-05-15' , freq = 'D' )
1783
- msg = 'Input has different freq from PeriodIndex\\ (freq=D\\ )'
1784
- with tm .assertRaisesRegexp (ValueError , msg ):
1785
+ msg = 'Input has different freq(=.+)? from PeriodIndex\\ (freq=D\\ )'
1786
+ with tm .assertRaisesRegexp (period . IncompatibleFrequency , msg ):
1785
1787
rng - o
1786
1788
1787
1789
offsets = [pd .offsets .Hour (2 ), timedelta (hours = 2 ),
@@ -1801,10 +1803,10 @@ def test_sub_isub(self):
1801
1803
np .timedelta64 (30 , 's' )]:
1802
1804
rng = pd .period_range ('2014-01-01 10:00' , '2014-01-05 10:00' ,
1803
1805
freq = 'H' )
1804
- msg = 'Input has different freq from PeriodIndex\\ (freq=H\\ )'
1805
- with tm .assertRaisesRegexp (ValueError , msg ):
1806
+ msg = 'Input has different freq(=.+)? from PeriodIndex\\ (freq=H\\ )'
1807
+ with tm .assertRaisesRegexp (period . IncompatibleFrequency , msg ):
1806
1808
result = rng + delta
1807
- with tm .assertRaisesRegexp (ValueError , msg ):
1809
+ with tm .assertRaisesRegexp (period . IncompatibleFrequency , msg ):
1808
1810
rng += delta
1809
1811
1810
1812
# int
0 commit comments