在 Python 脚本上运行测试 - 我做错了什么？

Question

我有一个简单的文件程序（从函数中删除代码，因为它与问题无关）：

import random
import sys


def poziom(level):
    *does something and returns something*

def gra(chance):
    *does something and returns something*


def save(name):
    *does something and returns something*


while True:
    *does something in loop, executing other functions above*

我正在尝试针对其功能之一运行一个简单的单元测试：

import unittest

from my_game import save
from my_game import gra
from my_game import poziom




class TestSum(unittest.TestCase):
    def test_list_int(self):
        """
        Test that it can sum a list of integers
        """
        check = "Test string"
        result = poziom(check)
        self.assertEqual(result, 0)

if __name__ == '__main__':
    unittest.main()

但是，此测试不是运行 poziom() 函数，而是运行整个文件，还执行 While True 语句后面的代码。 我做错了吗？在这种情况下是否可以编写一个只检查一个函数和它返回的值的测试？

Answer 1

但是，此测试不是运行 poziom() 函数，而是运行整个文件，还执行 While True 语句后面的代码。我做错了什么吗？

这就是import 的工作原理。它执行您导入的文件中的代码。要解决此问题，请将 while 循环放入 if __name__ == '__main__': 中，就像您在测试文件中所做的那样。