import timeit
import os
import sys
if "PURE_PYTHON" in os.environ:
    from cobs.cobs import _cobs_py as cobs
else:
    from cobs import cobs

number = int(os.environ.get("ITERS", "100"))

if 'encode' in sys.argv:
    with open("vmlinuz", "rb") as f:
        ddata = f.read()

    dtime = timeit.timeit(lambda: cobs.encode(ddata), number=number) / number / len(ddata)
    print(f"encoding: {1 / (dtime * 1e9)} GB/s")
else:
    with open("evmlinuz", "rb") as f:
        edata = f.read()
    etime = timeit.timeit(lambda: cobs.decode(edata), number=number) / number / len(edata)
    print(f"decoding: {1 / (etime * 1e9)} GB/s")