def _run_case(source):
from xdoctest import utils
COLOR = 'yellow'
def cprint(msg, color=COLOR):
print(utils.color_text(str(msg), COLOR))
cprint('\n\n' '\n <RUN CASE> ' '\n ======== ' '\n', COLOR)
cprint('DOCTEST SOURCE:')
cprint('---------------')
print(
utils.indent(
utils.add_line_numbers(utils.highlight_code(source, 'python'))))
print('')
import hashlib
hasher = hashlib.sha1()
hasher.update(source.encode('utf8'))
hashid = hasher.hexdigest()[0:8]
with utils.TempDir() as temp:
dpath = temp.dpath
modpath = join(dpath, 'test_linenos_' + hashid + '.py')
with open(modpath, 'w') as file:
file.write(source)
with utils.CaptureStdout(supress=False) as cap:
runner.doctest_module(modpath, 'all', argv=[''])
cprint('\n\n --- </END RUN CASE> --- \n\n', COLOR)
return cap.text
def _run_case(source, style='auto'):
"""
Runs all doctests in a source block
Args:
source (str): source code of an entire file
TODO: run case is over-duplicated and should be separated into a test utils directory
"""
from xdoctest import utils
from xdoctest import runner
COLOR = 'yellow'
def cprint(msg, color=COLOR):
print(utils.color_text(str(msg), COLOR))
cprint('\n\n'
'\n <RUN CASE> '
'\n ======== '
'\n', COLOR)
cprint('CASE SOURCE:')
cprint('------------')
print(utils.indent(
utils.add_line_numbers(utils.highlight_code(source, 'python'))))
print('')
import hashlib
hasher = hashlib.sha1()
hasher.update(source.encode('utf8'))
hashid = hasher.hexdigest()[0:8]
with utils.TempDir() as temp:
dpath = temp.dpath
modpath = join(dpath, 'test_linenos_' + hashid + '.py')
with open(modpath, 'w') as file:
file.write(source)
with utils.CaptureStdout(supress=False) as cap:
runner.doctest_module(modpath, 'all', argv=[''], style=style)
cprint('\n\n --- </END RUN CASE> --- \n\n', COLOR)
return cap.text
def format_part(self,
linenos=True,
want=True,
startline=1,
n_digits=None,
colored=False,
partnos=False,
prefix=True):
"""
Customizable formatting of the source and want for this doctest.
CommandLine:
python -m xdoctest.doctest_part DoctestPart.format_part
Args:
linenos (bool): show line numbers
want (bool): include the want value if it exists
startline (int): offsets the line numbering
n_digits (int): number of digits to use for line numbers
colored (bool): pygmentize the colde
partnos (bool): if True, shows the part number in the string
prefix (bool): if False, exclude the doctest `>>> ` prefix
CommandLine:
python -m xdoctest.doctest_part DoctestPart.format_part:0
Example:
>>> from xdoctest.parser import *
>>> self = DoctestPart(exec_lines=['print(123)'],
>>> want_lines=['123'], line_offset=0, partno=1)
>>> # xdoctest: -NORMALIZE_WHITESPACE
>>> print(self.format_part(partnos=True))
(p1) 1 >>> print(123)
123
Example:
>>> from xdoctest.parser import *
>>> self = DoctestPart(exec_lines=['print(123)'],
>>> want_lines=['123'], line_offset=0, partno=1)
>>> # xdoctest: -NORMALIZE_WHITESPACE
>>> print(self.format_part(partnos=False, prefix=False,
>>> linenos=False, want=False))
print(123)
"""
from xdoctest import utils
if prefix:
# Show the original line prefix when possible
if self.orig_lines is None:
src_text = utils.indent(self.source, '>>> ')
else:
src_text = '\n'.join(self.orig_lines)
else:
src_text = self.source
want_text = self.want if self.want else ''
if n_digits is None:
endline = startline + self.n_lines
n_digits = math.log(max(1, endline), 10)
n_digits = int(math.ceil(n_digits))
part_lines = src_text.splitlines()
n_spaces = 0
if linenos:
n_spaces += n_digits + 1
start = startline + self.line_offset
part_lines = utils.add_line_numbers(part_lines,
n_digits=n_digits,
start=start)
if partnos:
part_lines = [
'(p{}) {}'.format(self.partno, line) for line in part_lines
]
n_spaces += 4 + 1 # FIXME could be more robust if more than 9 parts
want_lines = []
if want_text:
want_fmt = ' ' * n_spaces + '{line}'
for line in want_text.splitlines():
if want:
want_lines.append(want_fmt.format(line=line))
part_text = '\n'.join(part_lines)
want_text = '\n'.join(want_lines)
if colored:
part_text = utils.highlight_code(part_text, 'python')
want_text = utils.color_text(want_text, 'green')
if want_lines:
part_text += '\n' + want_text
return part_text