From 69124cb6ab53f34e6dc7e93c7b9828c8e4a9b7ff Mon Sep 17 00:00:00 2001
From: Joe Jevnik <joejev@gmail.com>
Date: Mon, 13 Oct 2014 18:01:57 -0400
Subject: [PATCH] BUG: Fixed a bug with offsets in week_start and week_end

---
 tests/utils/test_events.py |  4 +--
 zipline/utils/events.py    | 59 +++++++++++++++++++++++++++++++-------
 2 files changed, 51 insertions(+), 12 deletions(-)

diff --git a/tests/utils/test_events.py b/tests/utils/test_events.py
index 7d8979f6..960c2979 100644
--- a/tests/utils/test_events.py
+++ b/tests/utils/test_events.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 import datetime
 import random
-from itertools import islice, dropwhile
+from itertools import islice
 from six.moves import range, map
 from nose_parameterized import parameterized
 from unittest import TestCase
@@ -281,7 +281,7 @@ class TestStatelessRules(RuleTestCase):
         should_trigger = NthTradingDayOfWeek(n).should_trigger
         prev_day = self.sept_week[0].date()
         n_tdays = 0
-        for m in dropwhile(lambda n: not should_trigger(n), self.sept_week):
+        for m in self.sept_week:
             if prev_day < m.date():
                 n_tdays += 1
             prev_day = m.date()
diff --git a/zipline/utils/events.py b/zipline/utils/events.py
index 1bd42a2b..ab2e12be 100644
--- a/zipline/utils/events.py
+++ b/zipline/utils/events.py
@@ -68,6 +68,45 @@ def ensure_utc(time, tz='UTC'):
     return time.replace(tzinfo=pytz.utc)
 
 
+def _coerce_datetime(maybe_dt):
+    if isinstance(maybe_dt, datetime.datetime):
+        return maybe_dt
+    elif isinstance(maybe_dt, datetime.date):
+        return datetime.datetime(
+            year=maybe_dt.year,
+            month=maybe_dt.month,
+            day=maybe_dt.day,
+            tzinfo=pytz.utc,
+        )
+    elif isinstance(maybe_dt, (tuple, list)) and len(maybe_dt) == 3:
+        year, month, day = maybe_dt
+        return datetime.datetime(
+            year=year,
+            month=month,
+            day=day,
+            tzinfo=pytz.utc,
+        )
+    else:
+        raise TypeError('Cannot coerce %s into a datetime.datetime'
+                        % type(maybe_dt).__name__)
+
+
+def _out_of_range_error(a, b=None, var='offset'):
+    start = 0
+    if b is None:
+        end = a - 1
+    else:
+        start = a
+        end = b - 1
+    return ValueError(
+        '{var} must be in between {start} and {end} inclusive'.format(
+            var=var,
+            start=start,
+            end=end,
+        )
+    )
+
+
 def _build_offset(offset, kwargs, default):
     """
     Builds the offset argument for event rules.
@@ -309,14 +348,14 @@ class NthTradingDayOfWeek(StatelessRule):
     """
     def __init__(self, n=0):
         if n not in range(5):
-            raise ValueError('n must be in [0,5)')
+            raise _out_of_range_error(5)
         self.td_delta = n
 
     def should_trigger(self, dt):
-        return self.env.add_trading_days(
+        return _coerce_datetime(self.env.add_trading_days(
             self.td_delta,
             self.get_first_trading_day_of_week(dt),
-        ) == dt.date()
+        )).date() == dt.date()
 
     def get_first_trading_day_of_week(self, dt):
         prev = dt
@@ -333,15 +372,15 @@ class NDaysBeforeLastTradingDayOfWeek(StatelessRule):
     """
     def __init__(self, n):
         if n not in range(5):
-            raise ValueError('n must be in [0,5)')
+            raise _out_of_range_error(5)
         self.td_delta = -n
         self.date = None
 
     def should_trigger(self, dt):
-        return self.env.add_trading_days(
+        return _coerce_datetime(self.env.add_trading_days(
             self.td_delta,
             self.get_last_trading_day_of_week(dt),
-        ) == dt.date()
+        )).date() == dt.date()
 
     def get_last_trading_day_of_week(self, dt):
         prev = dt
@@ -360,8 +399,8 @@ class NthTradingDayOfMonth(StatelessRule):
     This is zero-indexed, n=0 is the first trading day of the month.
     """
     def __init__(self, n=0):
-        if n not in range(31):
-            raise ValueError('n must be in [0,31)')
+        if n not in range(26):
+            raise _out_of_range_error(26)
         self.td_delta = n
         self.month = None
         self.day = None
@@ -398,8 +437,8 @@ class NDaysBeforeLastTradingDayOfMonth(StatelessRule):
     A rule that triggers n days before the last trading day of the month.
     """
     def __init__(self, n=0):
-        if n not in range(31):
-            raise ValueError('n must be in [0,31)')
+        if n not in range(26):
+            raise _out_of_range_error(26)
         self.td_delta = -n
         self.month = None
         self.day = None