56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
""" Doctests for NumPy-specific nose/doctest modifications
|
|
|
|
"""
|
|
from __future__ import division, absolute_import, print_function
|
|
|
|
# try the #random directive on the output line
|
|
def check_random_directive():
|
|
'''
|
|
>>> 2+2
|
|
<BadExample object at 0x084D05AC> #random: may vary on your system
|
|
'''
|
|
|
|
# check the implicit "import numpy as np"
|
|
def check_implicit_np():
|
|
'''
|
|
>>> np.array([1,2,3])
|
|
array([1, 2, 3])
|
|
'''
|
|
|
|
# there's some extraneous whitespace around the correct responses
|
|
def check_whitespace_enabled():
|
|
'''
|
|
# whitespace after the 3
|
|
>>> 1+2
|
|
3
|
|
|
|
# whitespace before the 7
|
|
>>> 3+4
|
|
7
|
|
'''
|
|
|
|
def check_empty_output():
|
|
""" Check that no output does not cause an error.
|
|
|
|
This is related to nose bug 445; the numpy plugin changed the
|
|
doctest-result-variable default and therefore hit this bug:
|
|
http://code.google.com/p/python-nose/issues/detail?id=445
|
|
|
|
>>> a = 10
|
|
"""
|
|
|
|
def check_skip():
|
|
""" Check skip directive
|
|
|
|
The test below should not run
|
|
|
|
>>> 1/0 #doctest: +SKIP
|
|
"""
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# Run tests outside numpy test rig
|
|
import nose
|
|
from numpy.testing.noseclasses import NumpyDoctest
|
|
argv = ['', __file__, '--with-numpydoctest']
|
|
nose.core.TestProgram(argv=argv, addplugins=[NumpyDoctest()])
|