Jupyter Notebook Python测试代码报错及解决方案
在使用Jupyter Notebook运行Python测试代码时，可能会遇到AttributeError: module '__main__' has no attribute '...'的错误。 这通常是因为你的测试代码的组织方式与Jupyter Notebook的环境不兼容导致的。
错误原因及解决方法：
该错误通常发生在测试代码直接调用了定义在Notebook中的变量或函数，而这些变量或函数在__main__模块中找不到。 解决方法是在测试脚本中使用if __name__ == '__main__':代码块来控制测试的执行。
立即学习“Python免费学习笔记（深入）”；
改进后的代码示例：
假设你的测试代码如下（错误示例）：
import unittest
def my_function():
return 10
class MyTest(unittest.TestCase):
def test_my_function(self):
self.assertEqual(my_function(), 10)
# 直接调用unittest.main()，导致错误
unittest.main()
修改后的代码如下（正确示例）：
import unittest
def my_function():
return 10
class MyTest(unittest.TestCase):
def test_my_function(self):
self.assertEqual(my_function(), 10)
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): 这行代码使用unittest模块运行测试套件。argv=['first-arg-is-ignored']是一个占位符参数，exit=False防止测试运行结束后Jupyter Notebook内核自动关闭。
通过这种方式，你的测试代码将只在Jupyter Notebook中直接运行时才执行测试，避免了__main__模块属性查找错误。 这使得你的测试代码更加健壮，也更适合在Jupyter Notebook等交互式环境中使用。
