~bzr-pqm/bzr/bzr.dev

1753.1.1 by Robert Collins
(rbc, jam, mbp)Add bzrlib.osutils.walkdirs, an optimised walk-and-stat routine.
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 bzr osutils functions performance."""
17
18
19
from bzrlib.benchmarks import Benchmark
20
import bzrlib.osutils as osutils
21
22
23
class WalkDirsBenchmark(Benchmark):
24
25
    def test_walkdirs_kernel_like_tree(self):
26
        """Walking a kernel sized tree is fast!(150ms)."""
27
        self.make_kernel_like_tree()
28
        # on roberts machine: this originally took:  157ms/4177ms
29
        # plain os.walk takes 213ms on this tree
30
        def dowalk():
31
            for dirblock in osutils.walkdirs('.'):
32
                if dirblock[0][1] == '.bzr':
33
                    del dirblock[0]
34
        self.time(dowalk)