~bzr-pqm/bzr/bzr.dev : contents of bzrlib/benchmarks/bench

~bzr-pqm/bzr/bzr.dev : (revision 4597.9.11)

# Copyright (C) 2007 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

"""Benchmarks for Knit performance"""

import os

from bzrlib import (
    generate_ids,
    knit,
    tests,
    )
from bzrlib.benchmarks import Benchmark
from bzrlib.tests.test_knit import CompiledKnitFeature


class BenchKnitIndex(Benchmark):
    """Benchmark Knit index performance."""

    def create_50k_index(self):
        """Create an knit index file with 50,000 entries.

        This isn't super realistic, but it *is* big :)

        The file 'test.kndx' will be created.
        """
        rev_id = generate_ids.gen_revision_id('long.name@this.example.com')
        versions = [(rev_id, ('fulltext',), 0, 200, [])]
        pos = 200
        alt_parent = None
        for i in xrange(49999):
            if alt_parent is not None:
                parent_ids = [rev_id, alt_parent]
            else:
                parent_ids = [rev_id]
            if i % 8 == 0:
                # The *next* entry will be a merge
                alt_parent = rev_id
            else:
                alt_parent = None
            rev_id = generate_ids.gen_revision_id('long.name@this.example.com')
            versions.append((rev_id, ('line-delta',), pos, 200, parent_ids))
            pos += 200
        t = self.get_transport()
        kndx = knit._KnitIndex(t, 'test.kndx', 'w', create=True,
                               delay_create=True)
        kndx.add_versions(versions)

    def setup_load_data_c(self):
        self.requireFeature(CompiledKnitFeature)
        from bzrlib._knit_load_data_pyx import _load_data_c
        self.overrideAttr(knit, '_load_data', _load_data_c)

    def setup_load_data_py(self):
        from bzrlib._knit_load_data_py import _load_data_py
        self.overrideAttr(knit, '_load_data', _load_data_py)

    def test_read_50k_index_c(self):
        self.setup_load_data_c()
        self.create_50k_index()
        t = self.get_transport()
        kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')

    def test_read_50k_index_c_again(self):
        self.setup_load_data_c()
        self.create_50k_index()
        t = self.get_transport()
        kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')

    def test_read_50k_index_py(self):
        self.create_50k_index()
        self.setup_load_data_py()
        t = self.get_transport()
        kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')

    def test_read_50k_index_py_again(self):
        self.create_50k_index()
        self.setup_load_data_py()
        t = self.get_transport()
        kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')

2484.1.2 by John Arbash Meinel Add benchmarks for loading a knit index with a lot of records.	1	# Copyright (C) 2007 Canonical Ltd
	2	#
	3	# This program is free software; you can redistribute it and/or modify
	4	# it under the terms of the GNU General Public License as published by
	5	# the Free Software Foundation; either version 2 of the License, or
	6	# (at your option) any later version.
	7	#
	8	# This program is distributed in the hope that it will be useful,
	9	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	10	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	11	# GNU General Public License for more details.
	12	#
	13	# You should have received a copy of the GNU General Public License
	14	# along with this program; if not, write to the Free Software
4183.7.1 by Sabin Iacob update FSF mailing address	15	# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
2484.1.2 by John Arbash Meinel Add benchmarks for loading a knit index with a lot of records.	16
	17	"""Benchmarks for Knit performance"""
	18
	19	import os
	20
	21	from bzrlib import (
	22	generate_ids,
	23	knit,
	24	tests,
	25	)
	26	from bzrlib.benchmarks import Benchmark
	27	from bzrlib.tests.test_knit import CompiledKnitFeature
	28
	29
	30	class BenchKnitIndex(Benchmark):
	31	"""Benchmark Knit index performance."""
	32
2484.1.3 by John Arbash Meinel Change to 50k entries, since this puts us in the 1s ballpark	33	def create_50k_index(self):
	34	"""Create an knit index file with 50,000 entries.
2484.1.2 by John Arbash Meinel Add benchmarks for loading a knit index with a lot of records.	35
	36	This isn't super realistic, but it is big :)
	37
	38	The file 'test.kndx' will be created.
	39	"""
	40	rev_id = generate_ids.gen_revision_id('long.name@this.example.com')
	41	versions = [(rev_id, ('fulltext',), 0, 200, [])]
	42	pos = 200
2484.1.3 by John Arbash Meinel Change to 50k entries, since this puts us in the 1s ballpark	43	alt_parent = None
	44	for i in xrange(49999):
	45	if alt_parent is not None:
	46	parent_ids = [rev_id, alt_parent]
	47	else:
	48	parent_ids = [rev_id]
	49	if i % 8 == 0:
	50	# The next entry will be a merge
	51	alt_parent = rev_id
	52	else:
	53	alt_parent = None
2484.1.2 by John Arbash Meinel Add benchmarks for loading a knit index with a lot of records.	54	rev_id = generate_ids.gen_revision_id('long.name@this.example.com')
	55	versions.append((rev_id, ('line-delta',), pos, 200, parent_ids))
	56	pos += 200
	57	t = self.get_transport()
	58	kndx = knit._KnitIndex(t, 'test.kndx', 'w', create=True,
	59	delay_create=True)
	60	kndx.add_versions(versions)
	61
	62	def setup_load_data_c(self):
	63	self.requireFeature(CompiledKnitFeature)
4573.1.1 by Andrew Bennetts Fix imports for _knit_load_data_pyx, which was recently renamed.	64	from bzrlib._knit_load_data_pyx import _load_data_c
4985.1.5 by Vincent Ladeuil Deploying the new overrideAttr facility further reduces the complexity	65	self.overrideAttr(knit, '_load_data', _load_data_c)
2484.1.2 by John Arbash Meinel Add benchmarks for loading a knit index with a lot of records.	66
	67	def setup_load_data_py(self):
2484.1.14 by John Arbash Meinel Fix up the benchmarks (bench_knit.py) for the new function locations.	68	from bzrlib._knit_load_data_py import _load_data_py
4985.1.5 by Vincent Ladeuil Deploying the new overrideAttr facility further reduces the complexity	69	self.overrideAttr(knit, '_load_data', _load_data_py)
2484.1.2 by John Arbash Meinel Add benchmarks for loading a knit index with a lot of records.	70
2484.1.3 by John Arbash Meinel Change to 50k entries, since this puts us in the 1s ballpark	71	def test_read_50k_index_c(self):
2484.1.6 by John Arbash Meinel Use direct functions when possible, and avoid extra dict lookups.	72	self.setup_load_data_c()
2484.1.3 by John Arbash Meinel Change to 50k entries, since this puts us in the 1s ballpark	73	self.create_50k_index()
	74	t = self.get_transport()
	75	kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')
	76
	77	def test_read_50k_index_c_again(self):
2484.1.6 by John Arbash Meinel Use direct functions when possible, and avoid extra dict lookups.	78	self.setup_load_data_c()
2484.1.3 by John Arbash Meinel Change to 50k entries, since this puts us in the 1s ballpark	79	self.create_50k_index()
	80	t = self.get_transport()
	81	kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')
	82
	83	def test_read_50k_index_py(self):
	84	self.create_50k_index()
2484.1.2 by John Arbash Meinel Add benchmarks for loading a knit index with a lot of records.	85	self.setup_load_data_py()
	86	t = self.get_transport()
	87	kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')
	88
2484.1.3 by John Arbash Meinel Change to 50k entries, since this puts us in the 1s ballpark	89	def test_read_50k_index_py_again(self):
	90	self.create_50k_index()
2484.1.2 by John Arbash Meinel Add benchmarks for loading a knit index with a lot of records.	91	self.setup_load_data_py()
	92	t = self.get_transport()
	93	kndx = self.time(knit._KnitIndex, t, 'test.kndx', 'r')