Jupyter Notebook Python测试报错：AttributeError: module '__main__' has no attribute '...' 如何解决？-Python教程-PHP中文网

Jupyter Notebook Python测试报错：AttributeError: module '__main__' has no attribute '...' 如何解决？

DDD

发布： 2025-02-23 17:42:28

原创

405人浏览过

jupyter notebook python测试报错：attributeerror: module '__main__' has no attribute '...' 如何解决？

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等交互式环境中使用。

以上就是Jupyter Notebook Python测试报错：AttributeError: module '__main__' has no attribute '...' 如何解决？的详细内容，更多请关注php中文网其它相关文章！