通过 py.test 在类属性上使用 doctest

Question

我尝试在 py.test 中使用 doctest 测试一个属性，但我得到一个 NameError 说该类未定义。

这是一个小例子：

class ExampleClass:

    def __init__(self, attribute):
        self.attribute = attribute

    @property
    def attribute_squared(self):
        """
        Examples:
            >>> ExampleClass(attribute=2).attribute_squared
            4
        """
        return self.attribute ** 2

当我在 py.test 中运行它时：

Error
**********************************************************************
Line 3, in ExampleClass
Failed example:
    ExampleClass(attribute=2)
Exception raised:
    Traceback (most recent call last):
      File "C:\Program Files\JetBrains\PyCharm Community Edition 2017.1.5\helpers\pycharm\docrunner.py", line 140, in __run
        compileflags, 1), test.globs)
      File "<doctest ExampleClass[0]>", line 1, in <module>
        ExampleClass(attribute=2).attribute_squared
    NameError: name 'ExampleClass' is not defined

有没有办法测试类属性，或者我应该编写一个真正的测试？

Answer 1

最后我发现只有当我在 Pycharm 中右键单击/运行“Doctest attribute_squared”时测试才通过。 运行pytest时，测试通过。