heapbench.py

#!/usr/bin/env python

import benchmark
import heapq


class BenchHeapq(benchmark.Benchmark):
    def eachSetUp(self):
        self.size = 500

        self.big = range(500 * self.size)
        self.small = range(2 * self.size)

        heapq.heapify(self.big)
        heapq.heapify(self.small)

    def test_big_heappop(self):
        for _ in xrange(self.size):
            heapq.heappop(self.big)

    def test_small_heappop(self):
        for _ in xrange(self.size):
            heapq.heappop(self.small)

    def test_bigpop(self):
        for _ in xrange(self.size):
            self.big.pop()

    def test_smallpop(self):
        for _ in xrange(self.size):
            self.small.pop()


if __name__ == '__main__':
    benchmark.main(each=50, numberFormat="%.4g")