import timeit

import pickle
import json
import simplejson
import ujson
# import orjson
import rapidjson
import yajl
import yaml


def load_pickle(f):
    return pickle.load(f)


def load_json(f):
    return json.load(f)


def load_simplejson(f):
    return simplejson.load(f)


def load_ujson(f):
    return ujson.load(f)


# def load_orjson(f):
#     return orjson.loads(f)


def load_rapidjson(f):
    s = f.read()
    return rapidjson.loads(s)


def load_pyyaml(f):
    return yaml.load(f, Loader=yaml.BaseLoader)


def load_yajl(f):
    return yajl.load(f)


print("json:")
print(timeit.Timer('load_json(open("test.json"))', 'from __main__ import load_json').timeit())

print("simplejson:")
print(timeit.Timer('load_simplejson(open("test.json"))', 'from __main__ import load_simplejson').timeit())

print("ujson:")
print(timeit.Timer('load_ujson(open("test.json"))', 'from __main__ import load_ujson').timeit())

# print("orjson:")
# print(timeit.Timer('load_orjson(open("test.json"))', 'from __main__ import load_orjson').timeit())

print("rapidjson:")
print(timeit.Timer('load_rapidjson(open("test.json"))', 'from __main__ import load_rapidjson').timeit())

print("PyYAML:")
print(timeit.Timer('load_pyyaml(open("test.json"))', 'from __main__ import load_pyyaml').timeit())

print("yajl:")
print(timeit.Timer('load_yajl(open("test.json"))', 'from __main__ import load_yajl').timeit())

print("pickle:")
print(timeit.Timer('load_pickle(open("test.pickle", "rb"))', 'from __main__ import load_pickle').timeit())