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

TST: py3 compat doctests

Browse Source
This commit is contained in:
Joe Jevnik
2016-06-20 15:16:33 -04:00
parent 5925107052
commit efd7bf72c3
3 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/pipeline/visualize.py
+4 -4
View File
@@ -25,10 +25,10 @@ def delimit(delimiters, content):
"""
Surround `content` with the first and last characters of `delimiters`.
>>> delimit('[]', "foo")
u'[foo]'
>>> delimit('""', "foo")
u'"foo"'
>>> delimit('[]', "foo") # doctest: +SKIP
'[foo]'
>>> delimit('""', "foo") # doctest: +SKIP
'"foo"'
"""
if len(delimiters) != 2:
raise ValueError(
zipline/utils/cache.py
+1
View File
@@ -41,6 +41,7 @@ class CachedObject(namedtuple("_CachedObject", "value expires")):
>>> obj.unwrap(expires)
1
>>> obj.unwrap(expires + Timedelta('1 minute'))
... # doctest: +IGNORE_EXCEPTION_DETAIL
Traceback (most recent call last):
...
Expired: 2014-01-01 00:00:00+00:00
zipline/utils/input_validation.py
+15 -13
View File
@@ -217,10 +217,11 @@ def expect_dtypes(**named):
>>> foo(arange(3), 'foo')
(array([0, 1, 2]), 'foo')
>>> foo(arange(3, dtype=float), 'foo') # doctest: +NORMALIZE_WHITESPACE
... # doctest: +ELLIPSIS
Traceback (most recent call last):
...
TypeError: zipline.utils.input_validation.foo() expected a value with
dtype 'int64' for argument 'x', but got 'float64' instead.
TypeError: ...foo() expected a value with dtype 'int64' for argument 'x',
but got 'float64' instead.
"""
for name, type_ in iteritems(named):
if not isinstance(type_, (dtype, tuple)):
@@ -279,11 +280,11 @@ def expect_kinds(**named):
2
>>> foo(int32(2))
2
>>> foo(float32(2)) # doctest: +NORMALIZE_WHITESPACE
>>> foo(float32(2)) # doctest: +NORMALIZE_WHITESPACE +ELLIPSIS
Traceback (most recent call last):
...
TypeError: zipline.utils.input_validation.foo() expected a numpy object of
kind 'i' for argument 'x', but got 'f' instead.
TypeError: ...foo() expected a numpy object of kind 'i' for argument 'x',
but got 'f' instead.
"""
for name, kind in iteritems(named):
if not isinstance(kind, (str, tuple)):
@@ -339,11 +340,11 @@ def expect_types(*_pos, **named):
...
>>> foo(2, '3')
(2, '3')
>>> foo(2.0, '3') # doctest: +NORMALIZE_WHITESPACE
>>> foo(2.0, '3') # doctest: +NORMALIZE_WHITESPACE +ELLIPSIS
Traceback (most recent call last):
...
TypeError: zipline.utils.input_validation.foo() expected a value of type
int for argument 'x', but got float instead.
TypeError: ...foo() expected a value of type int for argument 'x',
but got float instead.
"""
if _pos:
raise TypeError("expect_types() only takes keyword arguments.")
@@ -466,11 +467,11 @@ def expect_element(*_pos, **named):
'A'
>>> foo('b')
'B'
>>> foo('c') # doctest: +NORMALIZE_WHITESPACE
>>> foo('c') # doctest: +NORMALIZE_WHITESPACE +ELLIPSIS
Traceback (most recent call last):
...
ValueError: zipline.utils.input_validation.foo() expected a value in
('a', 'b') for argument 'x', but got 'c' instead.
ValueError: ...foo() expected a value in ('a', 'b') for argument 'x',
but got 'c' instead.
Notes
-----
@@ -510,10 +511,11 @@ def expect_dimensions(**dimensions):
>>> foo(array([1, 1]), array([[1, 1], [2, 2]]))
2
>>> foo(array([1, 1]), array([1, 1])) # doctest: +NORMALIZE_WHITESPACE
... # doctest: +ELLIPSIS
Traceback (most recent call last):
...
ValueError: zipline.utils.input_validation.foo() expected a 2-D array for
argument 'y', but got a 1-D array instead.
ValueError: ...foo() expected a 2-D array for argument 'y',
but got a 1-D array instead.
"""
def _expect_dimension(expected_ndim):
def _check(func, argname, argvalue):