From 42532ec880c23e2dac548591d273fc87bedd2b0a Mon Sep 17 00:00:00 2001 From: Randy Barlow Date: Tue, 1 Mar 2016 19:05:47 -0500 Subject: [PATCH] Support did yesterday. fixes #79 --- did/base.py | 5 +++++ tests/test_base.py | 6 ++++++ 2 files changed, 11 insertions(+) diff --git a/did/base.py b/did/base.py index 3eda4739..66376cf2 100644 --- a/did/base.py +++ b/did/base.py @@ -273,6 +273,11 @@ def period(argument): until = Date("today") until.date += delta(days=1) period = "today" + elif "yesterday" in argument: + since = Date("yesterday") + until = Date("yesterday") + until.date += delta(days=1) + period = "yesterday" elif "year" in argument: if "last" in argument: since, until = Date.last_year() diff --git a/tests/test_base.py b/tests/test_base.py index 69e71ef2..1e5e21ad 100644 --- a/tests/test_base.py +++ b/tests/test_base.py @@ -53,6 +53,12 @@ def test_Date_period(): from did.base import Date today = did.base.TODAY did.base.TODAY = datetime.date(2015, 10, 3) + # yesterday + for argument in ["yesterday"]: + since, until, period = Date.period(argument) + assert unicode(since) == "2015-10-02" + assert unicode(until) == "2015-10-03" + assert period == "yesterday" # This week for argument in ["", "week", "this week"]: since, until, period = Date.period(argument)