wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-24 13:00:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

TST: fix doctests

Browse Source
This commit is contained in:
llllllllll
2015-12-09 11:22:13 -05:00
parent 8220d1ee86
commit 48536add73
2 changed files with 11 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/lib/adjustment.pyx
+8 -20
View File
@@ -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
zipline/utils/input_validation.py
+3 -3
View File
@@ -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):
...