time_asdf_index.py

from pyasdf import AsdfFile
import numpy as np
import time

for i in (10, 100, 1000, 10000):
    arrays = []
    for x in range(i):
        arrays.append(np.random.rand(32, 32))

    tree = {'arrays': arrays}

    ff = AsdfFile(tree).write_to('test.asdf')
    t = time.time()
    with AsdfFile.open('test.asdf') as ff:
        np.array(ff.tree['arrays'][i-1])
    t0 = time.time() - t

    ff = AsdfFile(tree).write_to('test_noindex.asdf', include_block_index=False)
    t = time.time()
    with AsdfFile.open('test_noindex.asdf') as ff:
        np.array(ff.tree['arrays'][i-1])
    t1 = time.time() - t

    print("{0}: Indexed: {1}, Unindexed: {2}".format(i, t0, t1))