~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/benchmarks/bench_transform.py

Fix benchmark test

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2006 by 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 version 2 as published by
 
5
# the Free Software Foundation.
 
6
#
 
7
# This program is distributed in the hope that it will be useful,
 
8
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
9
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
10
# GNU General Public License for more details.
 
11
#
 
12
# You should have received a copy of the GNU General Public License
 
13
# along with this program; if not, write to the Free Software
 
14
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
 
 
16
"""Tests for tree transform performance"""
 
17
 
 
18
import os
 
19
 
 
20
from bzrlib.benchmarks import Benchmark
 
21
from bzrlib.osutils import pathjoin
 
22
from bzrlib.transform import TreeTransform
 
23
from bzrlib.workingtree import WorkingTree
 
24
 
 
25
class TransformBenchmark(Benchmark):
 
26
 
 
27
    def test_canonicalize_path(self):
 
28
        """Canonicalizing paths should be fast.""" 
 
29
        self.make_kernel_like_tree()
 
30
        wt = WorkingTree.open('.')
 
31
        paths = []
 
32
        for dirpath, dirnames, filenames in os.walk('.'):
 
33
            paths.extend(pathjoin(dirpath, d) for d in dirnames)
 
34
            paths.extend(pathjoin(dirpath, f) for f in filenames)
 
35
        tt = TreeTransform(wt)
 
36
        self.time(self.canonicalize_paths, tt, paths)
 
37
        tt.finalize()
 
38
 
 
39
    def canonicalize_paths(self, tt, paths):
 
40
        for path in paths:
 
41
            tt.canonical_path(path)