pytest 有时需要与 python shell 不同的导入语句

Question

我对编程有点陌生，到目前为止，我的大多数程序都是单目录事务，但现在我想尝试更大的东西，但我在导入工作时遇到了麻烦。我使用 pytest 创建我的单元测试，并且我一直在我的 test_foo.py 文件中的单元测试定义之后放置独立的测试脚本。这工作正常，直到我添加一个 __init__.py 文件。

文件排列1

StackOverflow
 | card.py
 | test_card.py

其中 test_card.py 使用 import card as crd 导入卡片。

结果：以下命令都没有遇到 ImportError

$ python3 test_card.py
$ pytest

文件排列2

StackOverflow
 | __init__.py
 | card.py
 | test_card.py

结果：如果我保持导入语句不变，$ python3 test_card.py 仍然有效，但 $ pytest 遇到 ImportError。

如果我改用import StackOverflow.card as crd，pytest 将再次开始工作，但由于 ImportError，我无法将其作为脚本运行。

我意识到在这个例子中我不需要__init__.py 文件，但是它只是一个更大程序的一部分。也有人指出，第二个 import 语句是完全错误的。那么如何让 pytest 使用原始的 import 语句呢？

card.py 全文：

#Created by Patrick Cunfer
#2017-07-15


class Card(object):
    def __init__(self, value, suit):
        '''
        Sets the card's value and suit
        param value: An integer from 1 to 13
        param suit: A character representing a suit
        '''
        self.value = value
        self.suit = suit


    def __str__(self):
        return str(self.value) + " of " + str(self.suit)

test_card.py 全文

#Created by Patrick Cunfer
#2017-07-15


# First one breaks pytest, second one breaks shell
# import card as crd
# import StackOverflow.card as crd


def test_Card():
    test = crd.Card(5, 'S')
    assert test.value == 5
    assert test.suit == 'S'
    assert str(test) == "5 of S"
    del test


if __name__ == "__main__":
    #Unit test found a bug? Lets isolate it!

    print("Test 1")
    test = crd.Card(5, 'S')

    print(test.suit)
    print("Expect 'S'")
    print()

    #etc.

Answer 1

如果您想在您的情况下使用__init__.py，则必须将父目录的StackOverflow 路径附加到sys，如下所示：

import sys
project_dir = 'path to the parent dir of StackOverflow'
sys.path.append(project_dir)

然后你就可以使用import StackOverflow.card as crd

---

您当前的import StackOverflow.card as crd 语句意味着您在StackOverflow 目录本身中有另一个名为StackOverflow 的目录；导入语句假设您的目录结构如下所示（不是您的情况）：

StackOverflow
    | test_card.py
    | StackOverflow
        | __init__.py
        | card.py

无论如何，在您的情况下，您不需要使用__init__.py，因为您正在从同一目录中导入另一个脚本。 __init__.py 通常用于在目录外导入脚本。

在此link 中，您将找到有关如何使用__init__.py 文件的文档。