|
2 | 2 | import pickle
|
3 | 3 | import cPickle
|
4 | 4 | import pickletools
|
5 |
| -import copyreg |
| 5 | +import copy_reg |
6 | 6 |
|
7 | 7 | from test.test_support import TestFailed, have_unicode, TESTFN, \
|
8 | 8 | run_with_locale
|
@@ -44,21 +44,21 @@ class ExtensionSaver:
|
44 | 44 | # there is one).
|
45 | 45 | def __init__(self, code):
|
46 | 46 | self.code = code
|
47 |
| - if code in copyreg._inverted_registry: |
48 |
| - self.pair = copyreg._inverted_registry[code] |
49 |
| - copyreg.remove_extension(self.pair[0], self.pair[1], code) |
| 47 | + if code in copy_reg._inverted_registry: |
| 48 | + self.pair = copy_reg._inverted_registry[code] |
| 49 | + copy_reg.remove_extension(self.pair[0], self.pair[1], code) |
50 | 50 | else:
|
51 | 51 | self.pair = None
|
52 | 52 |
|
53 | 53 | # Restore previous registration for code.
|
54 | 54 | def restore(self):
|
55 | 55 | code = self.code
|
56 |
| - curpair = copyreg._inverted_registry.get(code) |
| 56 | + curpair = copy_reg._inverted_registry.get(code) |
57 | 57 | if curpair is not None:
|
58 |
| - copyreg.remove_extension(curpair[0], curpair[1], code) |
| 58 | + copy_reg.remove_extension(curpair[0], curpair[1], code) |
59 | 59 | pair = self.pair
|
60 | 60 | if pair is not None:
|
61 |
| - copyreg.add_extension(pair[0], pair[1], code) |
| 61 | + copy_reg.add_extension(pair[0], pair[1], code) |
62 | 62 |
|
63 | 63 | class C:
|
64 | 64 | def __cmp__(self, other):
|
@@ -690,14 +690,14 @@ def test_newobj_generic(self):
|
690 | 690 | self.assertEqual(B(x), B(y), detail)
|
691 | 691 | self.assertEqual(x.__dict__, y.__dict__, detail)
|
692 | 692 |
|
693 |
| - # Register a type with copyreg, with extension code extcode. Pickle |
| 693 | + # Register a type with copy_reg, with extension code extcode. Pickle |
694 | 694 | # an object of that type. Check that the resulting pickle uses opcode
|
695 | 695 | # (EXT[124]) under proto 2, and not in proto 1.
|
696 | 696 |
|
697 | 697 | def produce_global_ext(self, extcode, opcode):
|
698 | 698 | e = ExtensionSaver(extcode)
|
699 | 699 | try:
|
700 |
| - copyreg.add_extension(__name__, "MyList", extcode) |
| 700 | + copy_reg.add_extension(__name__, "MyList", extcode) |
701 | 701 | x = MyList([1, 2, 3])
|
702 | 702 | x.foo = 42
|
703 | 703 | x.bar = "hello"
|
|
0 commit comments