Fix benchmark extension module import (#428)

This commit is contained in:
Contextualist 2020-06-21 22:27:52 -04:00 committed by GitHub
parent c1b1a23f62
commit 3508ca524e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,7 +1,7 @@
from msgpack import fallback from msgpack import fallback
try: try:
from msgpack import _unpacker, _packer from msgpack import _cmsgpack
has_ext = True has_ext = True
except ImportError: except ImportError:
@ -17,14 +17,14 @@ def profile(name, func):
def simple(name, data): def simple(name, data):
if has_ext: if has_ext:
packer = _packer.Packer() packer = _cmsgpack.Packer()
profile("packing %s (ext)" % name, lambda: packer.pack(data)) profile("packing %s (ext)" % name, lambda: packer.pack(data))
packer = fallback.Packer() packer = fallback.Packer()
profile("packing %s (fallback)" % name, lambda: packer.pack(data)) profile("packing %s (fallback)" % name, lambda: packer.pack(data))
data = packer.pack(data) data = packer.pack(data)
if has_ext: if has_ext:
profile("unpacking %s (ext)" % name, lambda: _unpacker.unpackb(data)) profile("unpacking %s (ext)" % name, lambda: _cmsgpack.unpackb(data))
profile("unpacking %s (fallback)" % name, lambda: fallback.unpackb(data)) profile("unpacking %s (fallback)" % name, lambda: fallback.unpackb(data))