From 48536add737875f87ff492d436cf44867d301267 Mon Sep 17 00:00:00 2001
From: llllllllll <joe@quantopian.com>
Date: Wed, 9 Dec 2015 11:22:13 -0500
Subject: [PATCH] TST: fix doctests

---
 zipline/lib/adjustment.pyx        | 28 ++++++++--------------------
 zipline/utils/input_validation.py |  6 +++---
 2 files changed, 11 insertions(+), 23 deletions(-)

diff --git a/zipline/lib/adjustment.pyx b/zipline/lib/adjustment.pyx
index cd9a1703..8bdeee00 100644
--- a/zipline/lib/adjustment.pyx
+++ b/zipline/lib/adjustment.pyx
@@ -509,16 +509,10 @@ cdef class Datetime64Overwrite(Datetime64Adjustment):
     >>> import numpy as np; import pandas as pd
     >>> dts = pd.date_range('2014', freq='D', periods=9, tz='UTC')
     >>> arr = dts.values.reshape(3, 3)
-    >>> arr
-    array([['2013-12-31T19:00:00.000000000-0500',
-            '2014-01-01T19:00:00.000000000-0500',
-            '2014-01-02T19:00:00.000000000-0500'],
-           ['2014-01-03T19:00:00.000000000-0500',
-            '2014-01-04T19:00:00.000000000-0500',
-            '2014-01-05T19:00:00.000000000-0500'],
-           ['2014-01-06T19:00:00.000000000-0500',
-            '2014-01-07T19:00:00.000000000-0500',
-            '2014-01-08T19:00:00.000000000-0500']], dtype='datetime64[ns]')
+    >>> arr == np.datetime64(0, 'ns')
+    array([[False, False, False],
+           [False, False, False],
+           [False, False, False]], dtype=bool)
     >>> adj = Datetime64Overwrite(
     ...     first_row=1,
     ...     last_row=2,
@@ -527,16 +521,10 @@ cdef class Datetime64Overwrite(Datetime64Adjustment):
     ...     value=np.datetime64(0, 'ns'),
     ... )
     >>> adj.mutate(arr.view(np.int64))
-    >>> arr
-    array([['2013-12-31T19:00:00.000000000-0500',
-            '2014-01-01T19:00:00.000000000-0500',
-            '2014-01-02T19:00:00.000000000-0500'],
-           ['2014-01-03T19:00:00.000000000-0500',
-            '1969-12-31T19:00:00.000000000-0500',
-            '1969-12-31T19:00:00.000000000-0500'],
-           ['2014-01-06T19:00:00.000000000-0500',
-            '1969-12-31T19:00:00.000000000-0500',
-            '1969-12-31T19:00:00.000000000-0500']], dtype='datetime64[ns]')
+    >>> arr == np.datetime64(0, 'ns')
+    array([[False, False, False],
+           [False,  True,  True],
+           [False,  True,  True]], dtype=bool)
     """
     cpdef mutate(self, int64_t[:, :] data):
         cdef Py_ssize_t row, col
diff --git a/zipline/utils/input_validation.py b/zipline/utils/input_validation.py
index edfc0764..8ef0ed70 100644
--- a/zipline/utils/input_validation.py
+++ b/zipline/utils/input_validation.py
@@ -66,13 +66,13 @@ def expect_dtypes(*_pos, **named):
 
     Usage
     -----
-    >>> from numpy import dtype
-    >>> @expect_types(x=dtype(int))
+    >>> from numpy import dtype, arange
+    >>> @expect_dtypes(x=dtype(int))
     ... def foo(x, y):
     ...    return x, y
     ...
     >>> foo(arange(3), 'foo')
-    (2, '3')
+    (array([0, 1, 2]), 'foo')
     >>> foo(arange(3, dtype=float), 'foo')
     Traceback (most recent call last):
        ...