Python doctest with newline characters: inconsistent leading whitespace error

kalefranz picture kalefranz · Sep 12, 2013 · Viewed 7.4k times · Source

When writing python doctests, how does one properly introduce newline characters within a string in the test? Here's a simple example:

def remove_newlines(text):
    """
    >>> remove_newlines("line1 \n"
    ...                 "still line 1\r"
    ...                 "now line2 \n"
    ...                 "more line2\n")
    line1 still line1
    now line2 more line2
    """
    return text.replace('\n', '')

import doctest
doctest.run_docstring_examples(remove_newlines, globals())

The output of which is:

Traceback (most recent call last):
...
ValueError: line 3 of the docstring for NoName has inconsistent leading whitespace: '"'

Answer

kalefranz picture kalefranz · Sep 12, 2013

The docstring docs actually allude to the problem, but not entirely clearly.

A couple of other stackoverflow threads here and here were helpful, but not easy to find given my own search criteria.

Here is my actual solution:

def remove_CRs(text):
    r"""
    >>> output = remove_CRs("line1 \r"
    ...                     "still line1\n"
    ...                     "now line2 \r"
    ...                     "more line2\r")
    >>> print(output)
    line1 still line1
    now line2 more line2
    """
    return text.replace('\r', '')

import doctest
doctest.run_docstring_examples(remove_CRs, globals())

Three things have changed from the original code snippet:

  1. The docstring had to be a raw python string.
  2. I had to use print() on the function output.
  3. I had to get over my own confusion with the difference between \n and \r. (That one's on me.)

Hope this saves someone else the couple of hours I spent on this.