当我将 pickle 与 unittest 一起使用时出现错误。

我写了 3 个程序文件:

1. 对于要 pickle 的类，
2. 对于使用 #1 中的类的类，
3. #2 中测试类的单元测试。

具体代码分别如下。

#1。 ClassToPickle.py

import pickle
class ClassToPickle(object):
    def __init__(self, x):
        self.x = x
if __name__=="__main__":
    p = ClassToPickle(10)
    pickle.dump(p, open('10.pickle', 'w'))

#2。一些类.py

from ClassToPickle import ClassToPickle
import pickle

class SomeClass(object):
    def __init__(self):
        self.pickle = pickle.load(open("10.pickle", 'r'))
        self.x = self.pickle.x
        print self.x

if __name__ == "__main__":
    SomeClass()

#3。一些类测试.py

import unittest
from SomeClass import SomeClass
from ClassToPickle import ClassToPickle # REQUIRED_LINE

class SomeClassTest(unittest.TestCase):
    def testA(self):
        sc = SomeClass()
        self.assertEqual(sc.x, 10)

def main():
    unittest.main()

if __name__ == "__main__":
    main()

我先运行#1 程序来制作 pickle 文件。
然后，当我单独运行程序文件 #2(即输入“python SomeClass.py”)时，它就起作用了。
而且，当我单独运行程序 #3(即输入“python SomeClassTest.py”)时，它也能正常工作。

但是，当我在 eclipse+pydev 中将程序 #3 作为“单元测试”运行时，它会在下面返回一条错误消息。

======================================================================
ERROR: testA (SomeClassTest.SomeClassTest)
----------------------------------------------------------------------
Traceback (most recent call last):
$ File "/home/tmp/pickle_problem/SomeClassTest.py", line 9, in testA
sc = SomeClass()
$ File "/home/tmp/pickle_problem/SomeClass.py", line 8, in init
self.pickle = pickle.load(open("10.pickle", 'r'))
$ File "/usr/lib/python2.7/pickle.py", line 1378, in load
return Unpickler(file).load()
$ File "/usr/lib/python2.7/pickle.py", line 858, in load
dispatchkey
File "/usr/lib/python2.7/pickle.py", line 1090, in load_global
klass = self.find_class(module, name)
$ File "/usr/lib/python2.7/pickle.py", line 1126, in find_class
klass = getattr(mod, name)
$ AttributeError: 'module' object has no attribute 'ClassToPickle'

----------------------------------------------------------------------
Ran 1 test in 0.002s

FAILED (errors=1)

而且，当我注释掉导入 ClassToPickle 类的行(程序 #3 中的第 3 行并注释为“REQUIRED_LINE”)时，它不起作用并返回如下所述的错误消息。

E
======================================================================
ERROR: testA (main.SomeClassTest)
----------------------------------------------------------------------
Traceback (most recent call last):
File "SomeClassTest.py", line 9, in testA
sc = SomeClass()
File "/home/tmp/pickle_problem/SomeClass.py", line 8, in init
self.pickle = pickle.load(open("10.pickle", 'r'))
File "/usr/lib/python2.7/pickle.py", line 1378, in load
return Unpickler(file).load()
File "/usr/lib/python2.7/pickle.py", line 858, in load
dispatchkey
File "/usr/lib/python2.7/pickle.py", line 1090, in load_global
klass = self.find_class(module, name)
File "/usr/lib/python2.7/pickle.py", line 1126, in find_class
klass = getattr(mod, name)
AttributeError: 'module' object has no attribute 'ClassToPickle'

----------------------------------------------------------------------
Ran 1 test in 0.001s

FAILED (errors=1)

我猜问题出在 python 中的命名空间，但我不知道到底发生了什么，也不知道我能做些什么来解决它。

如何“作为单元测试运行(在 eclipse+pydev 中)”#3 程序正确，
并在没有导入 ClassToPickle 的行的命令行中运行 #3 程序？
请帮助我。

最佳答案

那是因为 __main__.ClassToPickle != ClassToPickle.ClassToPickle，可以这样想:

当您在 ClassToPickle.py 脚本中 pickle ClassToPickle 的类实例时，pickle 模块将 pickle 所有对该类的引用，这意味着它将 pickle 定义该类的模块名称，因为你执行了脚本 ClassToPickle.py 这意味着模块将被设置为 __main__ 这意味着 pickle 模块将 pickle __main__.ClassToPickle.

当您尝试加载 pickled 实例时，它失败了，因为它没有找到实例的类 __main__.ClassToPickle 而不是您使用 from ClassToPickle import ClassToPickle 导入的类 因为这个最新的是 ClassToPickle.ClassToPickle。

解决方法是创建另一个脚本来处理转储，而不是在 ClassToPickle.py 中执行转储，例如

import pickle

from ClassToPickle import ClassToPickle

if __name__=="__main__":
    p = ClassToPickle(10)
    pickle.dump(p, open('10.pickle', 'w'))

关于python - python pickle 上的命名空间，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/7928450/