Skip to content

CLN: ASV offset #18926

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 3 commits into from
Dec 31, 2017
Merged

CLN: ASV offset #18926

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
57d16e5
CLN: ASV offset
mroeschke Dec 24, 2017
e97e7c0
Reduce samples for addition
mroeschke Dec 31, 2017
f7952bd
Do not lint pandas_vb_common.py yet
mroeschke Dec 31, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
266 changes: 65 additions & 201 deletions asv_bench/benchmarks/offset.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,257 +2,121 @@
from datetime import datetime

import numpy as np

import pandas as pd
from pandas import date_range

try:
import pandas.tseries.holiday
import pandas.tseries.holiday # noqa
except ImportError:
pass

hcal = pd.tseries.holiday.USFederalHolidayCalendar()
# These offests currently raise a NotImplimentedError with .apply_index()
non_apply = [pd.offsets.Day(),
pd.offsets.BYearEnd(),
pd.offsets.BYearBegin(),
pd.offsets.BQuarterEnd(),
pd.offsets.BQuarterBegin(),
pd.offsets.BMonthEnd(),
pd.offsets.BMonthBegin(),
pd.offsets.CustomBusinessDay(),
pd.offsets.CustomBusinessDay(calendar=hcal),
pd.offsets.CustomBusinessMonthBegin(calendar=hcal),
pd.offsets.CustomBusinessMonthEnd(calendar=hcal),
pd.offsets.CustomBusinessMonthEnd(calendar=hcal)]
other_offsets = [pd.offsets.YearEnd(), pd.offsets.YearBegin(),
pd.offsets.QuarterEnd(), pd.offsets.QuarterBegin(),
pd.offsets.MonthEnd(), pd.offsets.MonthBegin(),
pd.offsets.DateOffset(months=2, days=2),
pd.offsets.BusinessDay(), pd.offsets.SemiMonthEnd(),
pd.offsets.SemiMonthBegin()]
offsets = non_apply + other_offsets


class ApplyIndex(object):
goal_time = 0.2

params = [pd.offsets.YearEnd(), pd.offsets.YearBegin(),
pd.offsets.BYearEnd(), pd.offsets.BYearBegin(),
pd.offsets.QuarterEnd(), pd.offsets.QuarterBegin(),
pd.offsets.BQuarterEnd(), pd.offsets.BQuarterBegin(),
pd.offsets.MonthEnd(), pd.offsets.MonthBegin(),
pd.offsets.BMonthEnd(), pd.offsets.BMonthBegin()]

def setup(self, param):
self.offset = param
goal_time = 0.2

self.N = 100000
self.rng = date_range(start='1/1/2000', periods=self.N, freq='T')
self.ser = pd.Series(self.rng)
params = other_offsets
param_names = ['offset']

def time_apply_index(self, param):
self.rng + self.offset
def setup(self, offset):
N = 10000
self.rng = pd.date_range(start='1/1/2000', periods=N, freq='T')

def time_apply_series(self, param):
self.ser + self.offset
def time_apply_index(self, offset):
offset.apply_index(self.rng)


class OnOffset(object):

goal_time = 0.2

params = [pd.offsets.QuarterBegin(), pd.offsets.QuarterEnd(),
pd.offsets.BQuarterBegin(), pd.offsets.BQuarterEnd()]
params = offsets
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

def setup(self):
self.N = 100000
self.rng = date_range(start='1/1/2000', periods=self.N, freq='T')
self.day_offset = pd.offsets.Day()
self.relativedelta_offset = pd.offsets.DateOffset(months=2, days=2)
self.busday_offset = pd.offsets.BusinessDay()

def time_add_offset_delta(self):
self.rng + self.day_offset

def time_add_offset_fast(self):
self.rng + self.relativedelta_offset

def time_add_offset_slow(self):
self.rng + self.busday_offset


class SeriesArithmetic(object):
goal_time = 0.2
offset.onOffset(date)

def setup(self):
self.N = 100000
rng = date_range(start='20140101', freq='T', periods=self.N)
self.ser = pd.Series(rng)
self.day_offset = pd.offsets.Day()
self.relativedelta_offset = pd.offsets.DateOffset(months=2, days=2)
self.busday_offset = pd.offsets.BusinessDay()

def time_add_offset_delta(self):
self.ser + self.day_offset
class OffsetSeriesArithmetic(object):

def time_add_offset_fast(self):
self.ser + self.relativedelta_offset

def time_add_offset_slow(self):
self.ser + self.busday_offset


class YearBegin(object):
goal_time = 0.2
params = offsets
param_names = ['offset']

def setup(self):
self.date = datetime(2011, 1, 1)
self.year = pd.offsets.YearBegin()
def setup(self, offset):
N = 1000
rng = pd.date_range(start='1/1/2000', periods=N, freq='T')
self.data = pd.Series(rng)

def time_timeseries_year_apply(self):
self.year.apply(self.date)
def time_add_offset(self, offset):
self.data + offset

def time_timeseries_year_incr(self):
self.date + self.year

class OffsetDatetimeIndexArithmetic(object):

class Day(object):
goal_time = 0.2
params = offsets
param_names = ['offset']

def setup(self):
self.date = datetime(2011, 1, 1)
self.day = pd.offsets.Day()
def setup(self, offset):
N = 1000
self.data = pd.date_range(start='1/1/2000', periods=N, freq='T')

def time_timeseries_day_apply(self):
self.day.apply(self.date)
def time_add_offset(self, offset):
self.data + offset

def time_timeseries_day_incr(self):
self.date + self.day

class OffestDatetimeArithmetic(object):

class CBDay(object):
goal_time = 0.2
params = offsets
param_names = ['offset']

def setup(self):
def setup(self, offset):
self.date = datetime(2011, 1, 1)
self.dt64 = np.datetime64('2011-01-01 09:00Z')
self.cday = pd.offsets.CustomBusinessDay()

def time_custom_bday_decr(self):
self.date - self.cday

def time_custom_bday_incr(self):
self.date + self.cday

def time_custom_bday_apply(self):
self.cday.apply(self.date)

def time_custom_bday_apply_dt64(self):
self.cday.apply(self.dt64)


class CBDayHolidays(object):
goal_time = 0.2

def setup(self):
self.date = datetime(2011, 1, 1)
self.cdayh = pd.offsets.CustomBusinessDay(calendar=hcal)

def time_custom_bday_cal_incr(self):
self.date + 1 * self.cdayh

def time_custom_bday_cal_decr(self):
self.date - 1 * self.cdayh

def time_custom_bday_cal_incr_n(self):
self.date + 10 * self.cdayh

def time_custom_bday_cal_incr_neg_n(self):
self.date - 10 * self.cdayh


class CBMonthBegin(object):
goal_time = 0.2

def setup(self):
self.date = datetime(2011, 1, 1)
self.cmb = pd.offsets.CustomBusinessMonthBegin(calendar=hcal)

def time_custom_bmonthbegin_decr_n(self):
self.date - (10 * self.cmb)

def time_custom_bmonthbegin_incr_n(self):
self.date + (10 * self.cmb)


class CBMonthEnd(object):
goal_time = 0.2

def setup(self):
self.date = datetime(2011, 1, 1)
self.cme = pd.offsets.CustomBusinessMonthEnd(calendar=hcal)

def time_custom_bmonthend_incr(self):
self.date + self.cme

def time_custom_bmonthend_incr_n(self):
self.date + (10 * self.cme)

def time_custom_bmonthend_decr_n(self):
self.date - (10 * self.cme)


class SemiMonthOffset(object):
goal_time = 0.2

def setup(self):
self.N = 100000
self.rng = date_range(start='1/1/2000', periods=self.N, freq='T')
# date is not on an offset which will be slowest case
self.date = datetime(2011, 1, 2)
self.semi_month_end = pd.offsets.SemiMonthEnd()
self.semi_month_begin = pd.offsets.SemiMonthBegin()

def time_end_apply(self):
self.semi_month_end.apply(self.date)

def time_end_incr(self):
self.date + self.semi_month_end

def time_end_incr_n(self):
self.date + 10 * self.semi_month_end

def time_end_decr(self):
self.date - self.semi_month_end

def time_end_decr_n(self):
self.date - 10 * self.semi_month_end

def time_end_apply_index(self):
self.semi_month_end.apply_index(self.rng)

def time_end_incr_rng(self):
self.rng + self.semi_month_end

def time_end_decr_rng(self):
self.rng - self.semi_month_end

def time_begin_apply(self):
self.semi_month_begin.apply(self.date)

def time_begin_incr(self):
self.date + self.semi_month_begin

def time_begin_incr_n(self):
self.date + 10 * self.semi_month_begin
def time_apply(self, offset):
offset.apply(self.date)

def time_begin_decr(self):
self.date - self.semi_month_begin
def time_apply_np_dt64(self, offset):
offset.apply(self.dt64)

def time_begin_decr_n(self):
self.date - 10 * self.semi_month_begin
def time_add(self, offset):
self.date + offset

def time_begin_apply_index(self):
self.semi_month_begin.apply_index(self.rng)
def time_add_10(self, offset):
self.date + (10 * offset)

def time_begin_incr_rng(self):
self.rng + self.semi_month_begin
def time_subtract(self, offset):
self.date - offset

def time_begin_decr_rng(self):
self.rng - self.semi_month_begin
def time_subtract_10(self, offset):
self.date - (10 * offset)
2 changes: 1 addition & 1 deletion ci/lint.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ if [ "$LINT" ]; then
echo "Linting setup.py DONE"

echo "Linting asv_bench/benchmarks/"
flake8 asv_bench/benchmarks/ --exclude=asv_bench/benchmarks/[ijoprs]*.py --ignore=F811
flake8 asv_bench/benchmarks/ --exclude=asv_bench/benchmarks/[ips]*.py --ignore=F811
if [ $? -ne "0" ]; then
RET=1
fi
Expand Down