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脚本的运行环境有所不同。
有效的解决方法
立即学习“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文件需要在同一目录下。
