Как вставить конечные пробелы в доктрине, чтобы он не терпел неудачу, даже когда фактический и ожидаемый результат выглядят одинаково?

Я пытаюсь сделать доктрину. Результаты "Ожидаемые" и "Полученные" идентичны, но мой доктритель все еще терпит неудачу. Это не так, потому что в распечатке есть пробелы после x-axis y-axis, которые я не включил в свою docstring. Как включить его? Когда я вставляю пробелы вручную и выполняю тест, он выполняется успешно, пока я держу курсор.

ось оси x по оси x ______________________ [курсор здесь]

Однако, если я запустил тест с моим курсором в другом месте, то конечные пробелы будут удалены и тест завершится с ошибкой.

Я знаю, что звучит очень странно, но это то, что есть!

Это код:

import pandas as pd
import doctest


class NewDataStructure(pd.DataFrame):
    """
    >>> arrays = [[1, 1, 2, 2], [10, 20, 10, 20]]
    >>> index = pd.MultiIndex.from_arrays(arrays, names=('x-axis', 'y-axis'))
    >>> data_input = {"Pressure (Pa)": [1+1j, 2+2j, 3+3j, 4+4j],
    ...               "Temperature": [1, 2, 3, 4]}
    >>> new_data_variable = NewDataStructure(data=data_input, index=index, title="Pressures and temperatures")
    >>> print new_data_variable
    New Data Structure Pressures and temperatures:
                   Pressure (Pa)  Temperature
    x-axis y-axis                            
    1      10             (1+1j)            1
           20             (2+2j)            2
    2      10             (3+3j)            3
           20             (4+4j)            4

    """
    def __init__(self, data, index, title):
        super(NewDataStructure, self).__init__(data=data, index=index)
        self.title = title

    def __str__(self):
        return "New Data Structure {}:\n{}".format(self.title, super(NewDataStructure, self).__str__())

doctest.testmod()

Ниже мой результат, когда он терпит неудачу. Даже здесь вы можете выбрать область после x-axis y-axis и определить, существуют ли пробелы или нет.

Failed example:
    print new_data_variable
Expected:
    New Data Structure Pressures and temperatures:
                   Pressure (Pa)  Temperature
    x-axis y-axis
    1      10             (1+1j)            1
           20             (2+2j)            2
    2      10             (3+3j)            3
           20             (4+4j)            4
Got:
    New Data Structure Pressures and temperatures:
                   Pressure (Pa)  Temperature
    x-axis y-axis                            
    1      10             (1+1j)            1
           20             (2+2j)            2
    2      10             (3+3j)            3
           20             (4+4j)            4

Ответ 1

Я нашел решение, используя нормализовать флаг пробела

поместите это в доктрине как

    >>> print new_data_variable # doctest: +NORMALIZE_WHITESPACE

или при вызове doctest

doctest.testmod( optionflags= doctest.NORMALIZE_WHITESPACE )