mirror of
https://github.com/wassname/catalyst.git
synced 2026-07-09 17:29:26 +08:00
Scott Sanderson
75138343ba
Causes doctest to not care about the module of the raised exception. This matters because Python3 includes the module in the formatted exception, but Python3 doesn't, so the doctest will always fail on one or the other without this flag.
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
from __future__ import print_function
|
|
import sys
|
|
import doctest
|
|
from unittest import TestCase
|
|
|
|
from zipline.lib import adjustment
|
|
from zipline.modelling import (
|
|
engine,
|
|
expression,
|
|
)
|
|
from zipline.utils import (
|
|
memoize,
|
|
test_utils,
|
|
)
|
|
|
|
|
|
class DoctestTestCase(TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
import pdb
|
|
# Workaround for the issue addressed by this (unmerged) PR to pdbpp:
|
|
# https://bitbucket.org/antocuni/pdb/pull-request/40/fix-ensure_file_can_write_unicode/diff # noqa
|
|
if '_pdbpp_path_hack' in pdb.__file__:
|
|
cls._skip = True
|
|
else:
|
|
cls._skip = False
|
|
cls.flags = doctest.REPORT_CDIFF | doctest.IGNORE_EXCEPTION_DETAIL
|
|
|
|
def _check_docs(self, module):
|
|
if self._skip:
|
|
# Printing this directly to __stdout__ so that it doesn't get
|
|
# captured by nose.
|
|
print("Warning: Skipping doctests for %s because "
|
|
"pdbpp is installed." % module.__name__, file=sys.__stdout__)
|
|
return
|
|
try:
|
|
doctest.testmod(module, verbose=True, raise_on_error=True)
|
|
except doctest.UnexpectedException as e:
|
|
raise e.exc_info[1]
|
|
|
|
def test_adjustment_docs(self):
|
|
self._check_docs(adjustment)
|
|
|
|
def test_expression_docs(self):
|
|
self._check_docs(expression)
|
|
|
|
def test_engine_docs(self):
|
|
self._check_docs(engine)
|
|
|
|
def test_memoize_docs(self):
|
|
self._check_docs(memoize)
|
|
|
|
def test_test_utils_docs(self):
|
|
self._check_docs(test_utils)
|