add tests and asv for newly implemented methods

Author: jbrockmendel

asv_bench/benchmarks/offset.py

@@ -38,6 +38,25 @@ def time_apply_series(self, param):
         self.ser + self.offset
 
 
+class OnOffset(object):
+    goal_time = 0.2
+
+    params = [pd.offsets.QuarterBegin(), pd.offsets.QuarterEnd(),
+              pd.offsets.BQuarterBegin(), pd.offsets.BQuarterEnd()]
+    param_names = ['offset']
+
+    def setup(self, offset):
+        self.offset = offset
+        self.dates = [datetime(2016, m, d)
+                      for m in [10, 11, 12]
+                      for d in [1, 2, 3, 28, 29, 30, 31]
+                      if not (m == 11 and d == 31)]
+
+    def time_on_offset(self, offset):
+        for date in self.dates:
+            self.offset.onOffset(date)
+
+
 class DatetimeIndexArithmetic(object):
     goal_time = 0.2
 

pandas/tests/tseries/offsets/test_yqm_offsets.py

@@ -6,6 +6,7 @@
 
 import pytest
 
+import pandas as pd
 from pandas import Timestamp
 from pandas import compat
 
@@ -32,6 +33,36 @@ def test_quarterly_dont_normalize():
         assert (result.time() == date.time())
 
 
+
+@pytest.mark.parametrize('offset', [MonthBegin(), MonthEnd(),
+                                    BMonthBegin(), BMonthEnd()])
+def test_apply_index(offset):
+    rng = pd.date_range(start='1/1/2000', periods=100000, freq='T')
+    ser = pd.Series(rng)
+
+    res = rng + offset
+    res_v2 = offset.apply_index(rng)
+    assert (res == res_v2).all()
+    assert res[0] == rng[0] + offset
+    assert res[-1] == rng[-1] + offset
+    res2 = ser + offset
+    # apply_index is only for indexes, not series, so no res2_v2
+    assert res2.iloc[0] == ser.iloc[0] + offset
+    assert res2.iloc[-1] == ser.iloc[-1] + offset
+
+
+@pytest.mark.parametrize('offset', [QuarterBegin(), QuarterEnd(),
+                                    BQuarterBegin(), BQuarterEnd()])
+def test_on_offset(offset):
+    dates = [datetime(2016, m, d)
+             for m in [10, 11, 12]
+             for d in [1, 2, 3, 28, 29, 30, 31] if not (m == 11 and d == 31)]
+    for date in dates:
+        res = offset.onOffset(date)
+        slow_version = date == (date + offset) - offset
+        assert res == slow_version
+
+
 # --------------------------------------------------------------------
 # Months