import hyperjson
from collections import namedtuple
from operator import attrgetter
Contender = namedtuple('Contender', 'name,dumps,loads')
def pytest_benchmark_group_stats(config, benchmarks, group_by):
result = {}
for bench in benchmarks:
engine, data_kind = bench['param'].split('-')
group = result.setdefault("%s: %s" % (data_kind, bench['group']), [])
group.append(bench)
return sorted(result.items())
def pytest_addoption(parser):
parser.addoption('--compare', action='store_true',
help='compare against other JSON engines')
contenders = []
contenders.append(Contender('hyperjson',
hyperjson.dumps,
hyperjson.loads))
try:
import simplejson
except ImportError:
pass
else:
contenders.append(Contender('simplejson',
simplejson.dumps,
simplejson.loads))
try:
import ujson
except ImportError:
pass
else:
contenders.append(Contender('ujson',
ujson.dumps,
ujson.loads))
try:
import yajl
except ImportError:
pass
else:
contenders.append(Contender('yajl',
yajl.dumps,
yajl.loads))
try:
import rapidjson
except ImportError:
pass
else:
contenders.append(Contender('rapidjson',
rapidjson.dumps,
rapidjson.loads))
try:
import orjson
except ImportError:
pass
else:
contenders.append(Contender('orjson',
orjson.dumps,
orjson.loads))
def pytest_generate_tests(metafunc):
if 'contender' in metafunc.fixturenames:
if metafunc.config.getoption('compare'):
metafunc.parametrize('contender', contenders,
ids=attrgetter('name'))
else:
metafunc.parametrize(
'contender', contenders[:1], ids=attrgetter('name'))