本文整理汇总了Python中copy_reg.add_extension方法的典型用法代码示例。如果您正苦于以下问题:Python copy_reg.add_extension方法的具体用法?Python copy_reg.add_extension怎么用?Python copy_reg.add_extension使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类copy_reg
的用法示例。
在下文中一共展示了copy_reg.add_extension方法的8个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: restore
# 需要导入模块: import copy_reg [as 别名]
# 或者: from copy_reg import add_extension [as 别名]
def restore(self):
code = self.code
curpair = copy_reg._inverted_registry.get(code)
if curpair is not None:
copy_reg.remove_extension(curpair[0], curpair[1], code)
pair = self.pair
if pair is not None:
copy_reg.add_extension(pair[0], pair[1], code)
示例2: produce_global_ext
# 需要导入模块: import copy_reg [as 别名]
# 或者: from copy_reg import add_extension [as 别名]
def produce_global_ext(self, extcode, opcode):
e = ExtensionSaver(extcode)
try:
copy_reg.add_extension(__name__, "MyList", extcode)
x = MyList([1, 2, 3])
x.foo = 42
x.bar = "hello"
# Dump using protocol 1 for comparison.
s1 = self.dumps(x, 1)
self.assertIn(__name__, s1)
self.assertIn("MyList", s1)
self.assertEqual(opcode_in_pickle(opcode, s1), False)
y = self.loads(s1)
self.assertEqual(list(x), list(y))
self.assertEqual(x.__dict__, y.__dict__)
# Dump using protocol 2 for test.
s2 = self.dumps(x, 2)
self.assertNotIn(__name__, s2)
self.assertNotIn("MyList", s2)
self.assertEqual(opcode_in_pickle(opcode, s2), True)
y = self.loads(s2)
self.assertEqual(list(x), list(y))
self.assertEqual(x.__dict__, y.__dict__)
finally:
e.restore()
示例3: test_add_extension
# 需要导入模块: import copy_reg [as 别名]
# 或者: from copy_reg import add_extension [as 别名]
def test_add_extension(self):
global obj
obj = object()
#The module is system defined module:random
copy_reg.add_extension(random,obj,100)
#The module is a custom mudole or the module argument is not a type of module
global mod
mod = imp.new_module('module')
sys.modules['argu1'] = mod
import argu1
copy_reg.add_extension(argu1,obj,1)
module = True
copy_reg.add_extension(module,obj,6)
# the value is zero or less than zero
module = "module"
self.assertRaises(ValueError,copy_reg.add_extension,module,obj,0)
self.assertRaises(ValueError,copy_reg.add_extension,module,object(),-987654)
# the key is already registered with code
self.assertRaises(ValueError,copy_reg.add_extension,argu1,object(),100)
# the code is already in use for key
self.assertRaises(ValueError,copy_reg.add_extension,random,obj,100009)
#TODO: @skip("multiple_execute")
示例4: test_remove_extension
# 需要导入模块: import copy_reg [as 别名]
# 或者: from copy_reg import add_extension [as 别名]
def test_remove_extension(self):
#delete extension
copy_reg.remove_extension(random,obj,100)
import argu1
copy_reg.remove_extension(argu1,obj,1)
module = True
copy_reg.remove_extension(module,obj,6)
#remove extension which has not been registed
self.assertRaises(ValueError,copy_reg.remove_extension,random,obj,2)
self.assertRaises(ValueError,copy_reg.remove_extension,random,object(),100)
self.assertRaises(ValueError,copy_reg.remove_extension,argu1,obj,1)
copy_reg.add_extension(argu1,obj,1)
self.assertRaises(ValueError,copy_reg.remove_extension,argu1,obj,0)
示例5: test_inverted_registry
# 需要导入模块: import copy_reg [as 别名]
# 或者: from copy_reg import add_extension [as 别名]
def test_inverted_registry(self):
copy_reg.add_extension('obj1','obj2',64)
#get
result = copy_reg._inverted_registry[64]
self.assertTrue(result == ('obj1','obj2'),
"The _inverted_registry attribute did not return the correct value")
#set
value = ('newmodule','newobj')
copy_reg._inverted_registry[10001] = value
result = copy_reg._inverted_registry[10001]
self.assertTrue(result == value,
"The setattr of _inverted_registry attribute failed")
示例6: produce_global_ext
# 需要导入模块: import copy_reg [as 别名]
# 或者: from copy_reg import add_extension [as 别名]
def produce_global_ext(self, extcode, opcode):
e = ExtensionSaver(extcode)
try:
copy_reg.add_extension(__name__, "MyList", extcode)
x = MyList([1, 2, 3])
x.foo = 42
x.bar = "hello"
# Dump using protocol 1 for comparison.
s1 = self.dumps(x, 1)
self.assert_(__name__ in s1)
self.assert_("MyList" in s1)
self.assertEqual(opcode_in_pickle(opcode, s1), False)
y = self.loads(s1)
self.assertEqual(list(x), list(y))
self.assertEqual(x.__dict__, y.__dict__)
# Dump using protocol 2 for test.
s2 = self.dumps(x, 2)
self.assert_(__name__ not in s2)
self.assert_("MyList" not in s2)
self.assertEqual(opcode_in_pickle(opcode, s2), True)
y = self.loads(s2)
self.assertEqual(list(x), list(y))
self.assertEqual(x.__dict__, y.__dict__)
finally:
e.restore()
示例7: test_extension_registry
# 需要导入模块: import copy_reg [as 别名]
# 或者: from copy_reg import add_extension [as 别名]
def test_extension_registry(self):
mod, func, code = 'junk1 ', ' junk2', 0xabcd
e = ExtensionSaver(code)
try:
# Shouldn't be in registry now.
self.assertRaises(ValueError, copy_reg.remove_extension,
mod, func, code)
copy_reg.add_extension(mod, func, code)
# Should be in the registry.
self.assertTrue(copy_reg._extension_registry[mod, func] == code)
self.assertTrue(copy_reg._inverted_registry[code] == (mod, func))
# Shouldn't be in the cache.
self.assertNotIn(code, copy_reg._extension_cache)
# Redundant registration should be OK.
copy_reg.add_extension(mod, func, code) # shouldn't blow up
# Conflicting code.
self.assertRaises(ValueError, copy_reg.add_extension,
mod, func, code + 1)
self.assertRaises(ValueError, copy_reg.remove_extension,
mod, func, code + 1)
# Conflicting module name.
self.assertRaises(ValueError, copy_reg.add_extension,
mod[1:], func, code )
self.assertRaises(ValueError, copy_reg.remove_extension,
mod[1:], func, code )
# Conflicting function name.
self.assertRaises(ValueError, copy_reg.add_extension,
mod, func[1:], code)
self.assertRaises(ValueError, copy_reg.remove_extension,
mod, func[1:], code)
# Can't remove one that isn't registered at all.
if code + 1 not in copy_reg._inverted_registry:
self.assertRaises(ValueError, copy_reg.remove_extension,
mod[1:], func[1:], code + 1)
finally:
e.restore()
# Shouldn't be there anymore.
self.assertNotIn((mod, func), copy_reg._extension_registry)
# The code *may* be in copy_reg._extension_registry, though, if
# we happened to pick on a registered code. So don't check for
# that.
# Check valid codes at the limits.
for code in 1, 0x7fffffff:
e = ExtensionSaver(code)
try:
copy_reg.add_extension(mod, func, code)
copy_reg.remove_extension(mod, func, code)
finally:
e.restore()
# Ensure invalid codes blow up.
for code in -1, 0, 0x80000000L:
self.assertRaises(ValueError, copy_reg.add_extension,
mod, func, code)
示例8: test_extension_registry
# 需要导入模块: import copy_reg [as 别名]
# 或者: from copy_reg import add_extension [as 别名]
def test_extension_registry(self):
mod, func, code = 'junk1 ', ' junk2', 0xabcd
e = ExtensionSaver(code)
try:
# Shouldn't be in registry now.
self.assertRaises(ValueError, copy_reg.remove_extension,
mod, func, code)
copy_reg.add_extension(mod, func, code)
# Should be in the registry.
self.assert_(copy_reg._extension_registry[mod, func] == code)
self.assert_(copy_reg._inverted_registry[code] == (mod, func))
# Shouldn't be in the cache.
self.assert_(code not in copy_reg._extension_cache)
# Redundant registration should be OK.
copy_reg.add_extension(mod, func, code) # shouldn't blow up
# Conflicting code.
self.assertRaises(ValueError, copy_reg.add_extension,
mod, func, code + 1)
self.assertRaises(ValueError, copy_reg.remove_extension,
mod, func, code + 1)
# Conflicting module name.
self.assertRaises(ValueError, copy_reg.add_extension,
mod[1:], func, code )
self.assertRaises(ValueError, copy_reg.remove_extension,
mod[1:], func, code )
# Conflicting function name.
self.assertRaises(ValueError, copy_reg.add_extension,
mod, func[1:], code)
self.assertRaises(ValueError, copy_reg.remove_extension,
mod, func[1:], code)
# Can't remove one that isn't registered at all.
if code + 1 not in copy_reg._inverted_registry:
self.assertRaises(ValueError, copy_reg.remove_extension,
mod[1:], func[1:], code + 1)
finally:
e.restore()
# Shouldn't be there anymore.
self.assert_((mod, func) not in copy_reg._extension_registry)
# The code *may* be in copy_reg._extension_registry, though, if
# we happened to pick on a registered code. So don't check for
# that.
# Check valid codes at the limits.
for code in 1, 0x7fffffff:
e = ExtensionSaver(code)
try:
copy_reg.add_extension(mod, func, code)
copy_reg.remove_extension(mod, func, code)
finally:
e.restore()
# Ensure invalid codes blow up.
for code in -1, 0, 0x80000000L:
self.assertRaises(ValueError, copy_reg.add_extension,
mod, func, code)