Doctest 和相关导入

Question

我在使用 doctest 和相对导入时遇到问题。简单的解决方案就是摆脱相对导入。还有其他人吗？

假设我有一个名为 example 的包，其中包含 2 个文件：

example/__init__.py

"""
This package is entirely useless.
>>> arnold = Aardvark()
>>> arnold.talk()
I am an aardvark.
"""

from .A import Aardvark

if __name__ == "__main__":
    import doctest
    doctest.testmod()

example/A.py

class Aardvark(object):
    def talk(self):
        print("I am an aardvark.")

如果我现在尝试

python example/__init__.py

，那么我会得到错误

Traceback (most recent call last):
  File "example/__init__.py", line 8, in <module>
    from .A import Aardvark
ValueError: Attempted relative import in non-package

Answer 1

创建另一个文件my_doctest_runner.py：

if __name__ == "__main__":
    import doctest
    import example
    doctest.testmod(example)

执行my_doctest_runner.py以运行example/__init__.py中的doctests：

$ python2.7 my_doctest_runner.py
**********************************************************************
File "/tmp/example/__init__.py", line 4, in example
Failed example:
    arnold.talk()
Expected:
    I am an aaardvark.
Got:
    I am an aardvark.
**********************************************************************
1 items had failures:
   1 of   2 in example
***Test Failed*** 1 failures.

Answer 2

Pytest 的 --doctest-modules< /a> 标志负责相对导入：

$ ls example/
A.py  __init__.py
$ pytest --doctest-modules example
==================== test session starts ====================
...

example/__init__.py .                                 [100%]

===================== 1 passed in 0.03s =====================

Answer 3

只要做

from A import Aardvark

Answer 4

从 pytest 6.0.0 开始，importlib 是有效的 导入模式。使用它解决了我的问题。

pytest --import-mode=importlib