# 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')
|