immutables benchmark

bench.py

import pyrsistent
import immutables
import time

I = 1_000_000
KEY = '5'


for N in [5, 10, 20, 30, 100, 200, 300, 400, 500, 1000]:
    print('=============')
    print(f'  # of items: {N}; iterations: {I}')
    print()


    h = immutables.Map()
    d = dict()
    pm = pyrsistent.m()

    for i in range(N):
        h = h.set(str(i), i)
        pm = pm.set(str(i), i)
        d[str(i)] = i

    assert len(h) == N

    for i in range(N):
        assert h.get(str(i), 'not found') == i


    st = time.monotonic()
    for _ in range(I):
        d.get(KEY)
        d.get(KEY)
        d.get(KEY)
        d.get(KEY)
        d.get(KEY)

        d.get(KEY)
        d.get(KEY)
        d.get(KEY)
        d.get(KEY)
        d.get(KEY)

        d2 = d.copy()
        d2['aaa'] = 'aaa'

    end = time.monotonic() - st
    print(f"  dict copy:\t\t\t{end:.4f}s")


    st = time.monotonic()
    for _ in range(I):
        h.get(KEY)
        h.get(KEY)
        h.get(KEY)
        h.get(KEY)
        h.get(KEY)

        h.get(KEY)
        h.get(KEY)
        h.get(KEY)
        h.get(KEY)
        h.get(KEY)

        h2 = h.set('aaa', 'aaa')

    end = time.monotonic() - st
    print(f"  hamt:\t\t\t\t{end:.4f}s")


    st = time.monotonic()
    for _ in range(I):
        pm.get(KEY)
        pm.get(KEY)
        pm.get(KEY)
        pm.get(KEY)
        pm.get(KEY)

        pm.get(KEY)
        pm.get(KEY)
        pm.get(KEY)
        pm.get(KEY)
        pm.get(KEY)

        pm2 = pm.set('aaa', 'aaa')

    end = time.monotonic() - st
    print(f"  PMap:\t\t\t\t{end:.4f}s")