导入本地模块的python pytest导入模块

Question

我有以下python3项目结构：

tests/
    - testsuite_service1/
        - test_main.py
    - testsuite_service2/
        - test_main.py
src/
    - service1/
       - codebase/
           - __init__.py
           - main.py
           - logger.py
           - waiter.py
    - service2/
       - codebase/
           - __init__.py
           - main.py
           - logger.py
           - waiter.py

在我的测试(pytest)中，我是这样导入的：

from src.service1.codebase.waiter import check_status

为了在 service1 模块中导入特定功能。

在waiter.py in service1 我正在从logger.py 导入一个函数，就像这样

from logger import configure_logger

但是，在运行测试时，我得到了错误：

Traceback:
tests/test_main/test_main.py:3: in <module>
    from src.service1.codebase.waiter import check_status
src/codebase/waiter.py:8: in <module>
    from logger import configure_logger
E   ModuleNotFoundError: No module named 'logger'.

这发生在测试运行时，感觉waiter.py 是不是在其本地目录中查找logger.py 而是可能在其他位置？

改成

from .logger import configure_logger

确实解决了这个问题，但会导致运行时出现问题，我计划将其部署到 （service1 的单个 zip，没有父目录），因此会出现错误 Unable to import module 'main': attempted relative import with no known parent package

是否可以不使用相对导入并让我的测试了解在哪里可以找到所需的文件？

Answer 1

包src 应该在根级别包含__init__.py 文件，以表明它是一个包。然后您可以使用绝对或相对路径从其他文件导入。绝对路径是

from src.service1.codebase.logger import configure_logger

但是在包含 __init__.py 文件之后，您的相对路径应该可以工作。

查看this answer了解更多信息。

Answer 2

在 codebase/ 文件夹中添加 import os, sys; sys.path.append(os.path.dirname(os.path.realpath(__file__))) 到 __init__.py 解决了这个问题。