add missing files

tzh

2024-08-22 c7d0944258c7d0943aa7b2211498fd612971ce27

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import pickle
from cStringIO import StringIO
 
from test import test_support
 
from test.pickletester import (AbstractPickleTests,
                               AbstractPickleModuleTests,
                               AbstractPersistentPicklerTests,
                               AbstractPicklerUnpicklerObjectTests,
                               BigmemPickleTests)
 
class PickleTests(AbstractPickleTests, AbstractPickleModuleTests):
 
    def dumps(self, arg, proto=0, fast=0):
        # Ignore fast
        return pickle.dumps(arg, proto)
 
    def loads(self, buf):
        # Ignore fast
        return pickle.loads(buf)
 
    module = pickle
    error = KeyError
 
class PicklerTests(AbstractPickleTests):
 
    error = KeyError
 
    def dumps(self, arg, proto=0, fast=0):
        f = StringIO()
        p = pickle.Pickler(f, proto)
        if fast:
            p.fast = fast
        p.dump(arg)
        f.seek(0)
        return f.read()
 
    def loads(self, buf):
        f = StringIO(buf)
        u = pickle.Unpickler(f)
        return u.load()
 
class PersPicklerTests(AbstractPersistentPicklerTests):
 
    def dumps(self, arg, proto=0, fast=0):
        class PersPickler(pickle.Pickler):
            def persistent_id(subself, obj):
                return self.persistent_id(obj)
        f = StringIO()
        p = PersPickler(f, proto)
        if fast:
            p.fast = fast
        p.dump(arg)
        f.seek(0)
        return f.read()
 
    def loads(self, buf):
        class PersUnpickler(pickle.Unpickler):
            def persistent_load(subself, obj):
                return self.persistent_load(obj)
        f = StringIO(buf)
        u = PersUnpickler(f)
        return u.load()
 
class PicklerUnpicklerObjectTests(AbstractPicklerUnpicklerObjectTests):
 
    pickler_class = pickle.Pickler
    unpickler_class = pickle.Unpickler
 
class PickleBigmemPickleTests(BigmemPickleTests):
 
    def dumps(self, arg, proto=0, fast=0):
        # Ignore fast
        return pickle.dumps(arg, proto)
 
    def loads(self, buf):
        # Ignore fast
        return pickle.loads(buf)
 
 
def test_main():
    test_support.run_unittest(
        PickleTests,
        PicklerTests,
        PersPicklerTests,
        PicklerUnpicklerObjectTests,
        PickleBigmemPickleTests,
    )
    test_support.run_doctest(pickle)
 
if __name__ == "__main__":
    test_main()