JupyterNotebook单元测试报错解决方法

Jupyter Notebook运行Python单元测试时，使用`unittest.main()`常常导致程序意外退出并报错（如`AttributeError`或`SystemExit`）。这是因为`unittest.main()`默认调用`sys.exit()`终止内核。本文提供解决方案：在测试脚本中添加`if __name__ == '__main__':`代码块，并使用`unittest.main(argv=['first-arg-is-ignored'], exit=False)`运行测试。 此修改可避免内核终止，让测试结果直接在Jupyter Notebook单元格中显示，有效解决`unittest.main()`导致的程序退出问题。

Jupyter Notebook运行Python单元测试报错及解决方案

在Jupyter Notebook中运行Python单元测试代码时，使用unittest.main()可能会导致程序意外退出，并出现AttributeError或SystemExit错误。本文将分析该问题，并提供有效的解决方案。

问题描述：

下图展示了在Jupyter Notebook中运行单元测试代码时遇到的错误。

代码示例：

name_function.py:

def name_function_name(first, last):
    """生成格式规范的完整姓名"""
    full_name = first + ' ' + last
    return full_name.title()

test_name_function.py:

import unittest
from name_function import name_function_name

class NameTestCase(unittest.TestCase):
    """测试name_function"""
    def test_first_last_name(self):
        formatted_name = name_function_name('janis', 'joplin')
        self.assertEqual(formatted_name, 'Janis Joplin')

unittest.main()

错误信息通常类似于：

AttributeError: module '__main__' has no attribute '...'
SystemExit: True

解决方案：

unittest.main()函数在执行完毕后会默认调用sys.exit()，这会导致Jupyter Notebook内核终止。为了避免这种情况，需要修改test_name_function.py文件，添加以下代码：

import unittest
from name_function import name_function_name

class NameTestCase(unittest.TestCase):
    """测试name_function"""
    def test_first_last_name(self):
        formatted_name = name_function_name('janis', 'joplin')
        self.assertEqual(formatted_name, 'Janis Joplin')

if __name__ == '__main__':
    unittest.main(argv=['first-arg-is-ignored'], exit=False)

关键修改：

• if __name__ == '__main__': 这行代码确保unittest.main()只在脚本直接运行时执行，而不是在作为模块导入时执行。
• unittest.main(argv=['first-arg-is-ignored'], exit=False) argv参数是为了避免unittest.main()对命令行参数的处理，exit=False阻止了unittest.main()调用sys.exit()。

通过以上修改，unittest.main()将不会终止Jupyter Notebook内核，测试结果将正常显示在Jupyter Notebook单元格中。 记住，assertEqual中的预期结果应为'Janis Joplin'，而不是'janis joplin'，因为title()方法会将首字母大写。

今天带大家了解了的相关知识，希望对你有所帮助；关于文章的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~