python-3.x - 停止单元测试从导入模块运行代码

Question

有 2 个文件:

new_project
├── Main.py
└── testing.py

我跑:

$ cd new_project
$ python -m unittest testing.py

我的整个测试文件是:

import unittest
from Main import Square


class TestSquare(unittest.TestCase):

    def test_init(self):
        self.assertEqual(Square(0,0).center, Point(50,50))
        print("test")

Square 是我的 Main.py 文件中的第一个类。Main.py 的组成如下:

import sys
import math
import random


def d_print(message):
    print(message, file=sys.stderr)


class Square:
    # on découpe le terrain en square et on les marque avec top right corner of square
    def __init__(self, x, y):
        self.xtr = x
        self.ytr = y
        self.size = 100
        self.explored = False
        self.center = Point(x+50, y+50)

while True:
    # do stuff every turn
    x, y, value = [int(j) for j in input().split()]

while 循环内的代码将由游戏模拟器每回合调用。模拟器提供输入。

当我运行 unittest 命令行时，它实际上是在等待输入

如果我删除 Square 上的 import Main 和 TestFixture，则单元测试通过。我尝试了几种配置，但在没有启动 while 循环的情况下，我无法设法导入 Square 进行测试。

所以当我只从 Main.py 导入一个类时，它仍然在类外运行代码。这让我很困惑。这是导入机制的链接，我不明白为什么它会运行代码以及如何防止它 https://docs.python.org/3/reference/import.html

由于游戏模拟不在我的控制范围内，所以我无法对 Main.py 的编写方式进行太多更改。我唯一的想法是将循环和类拆分为两个文件，用于开发和测试；提交时，我必须将文件连接回一个。

那么(感谢到目前为止的阅读；):

• 为什么 unittest/import 以这种方式工作？
• 知道如何解决吗？ (我现在正在尝试拆分文件的想法，会回来报告)

Answer 1

为避免运行循环，当文件作为模块导入时，这样写:

if __name__ == "__main__":
    while True:
        # do stuff every turn
        x, y, value = [int(j) for j in input().split()]

答案很简单，当您导入模块时，该文件中的代码将被执行，这将导致循环运行并在 input() 上阻塞执行。有关其工作原理的更多详细信息，请 read this answer with a detailed explanation