Jupyter Notebook中运行Python单元测试时出现AttributeError错误如何解决？

Jupyter Notebook中Python单元测试的AttributeError错误及解决方法

在Jupyter Notebook中运行Python单元测试时，初学者可能会遇到AttributeError: module '__main__' has no attribute '...'这样的错误。 例如，以下测试代码在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错误，这是因为Jupyter Notebook的运行环境与标准Python脚本的运行环境有所不同。

有效的解决方法

为了避免AttributeError错误，可以在测试脚本中添加以下代码块：

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

这段代码创建了一个新的命名空间来运行单元测试，从而避免与Jupyter Notebook的命名空间冲突。argv=['first-arg-is-ignored']参数确保unittest将命令行参数视为第一个参数（通常是文件名），而exit=False则阻止unittest自动退出，允许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__':代码块，可以确保测试代码在Jupyter Notebook中正确运行，并避免AttributeError错误。 记住，name_function.py文件需要在同一目录下。