about summary refs log tree commit diff
path: root/third_party/immer/extra/python/benchmark/test_benchmarks.py
blob: 11e4817c3256906fb7a3f4c586804a2f42d1b9ff (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

# immer: immutable data structures for C++
# Copyright (C) 2016, 2017, 2018 Juan Pedro Bolivar Puente
#
# This software is distributed under the Boost Software License, Version 1.0.
# See accompanying file LICENSE or copy at http://boost.org/LICENSE_1_0.txt

##

import immer
import pyrsistent

BENCHMARK_SIZE = 1000

def push(v, n=BENCHMARK_SIZE):
    for x in xrange(n):
        v = v.append(x)
    return v

def assoc(v):
    for i in xrange(len(v)):
        v = v.set(i, i+1)
    return v

def index(v):
    for i in xrange(len(v)):
        v[i]

def test_push_immer(benchmark):
    benchmark(push, immer.Vector())

def test_push_pyrsistent(benchmark):
    benchmark(push, pyrsistent.pvector())

def test_assoc_immer(benchmark):
    benchmark(assoc, push(immer.Vector()))

def test_assoc_pyrsistent(benchmark):
    benchmark(assoc, push(pyrsistent.pvector()))

def test_index_immer(benchmark):
    benchmark(index, push(immer.Vector()))

def test_index_pyrsistent(benchmark):
    benchmark(index, push(pyrsistent.pvector()))