← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/benchmarks/__init__.py

  • Committer: Aaron Bentley
  • Date: 2007-06-21 23:43:17 UTC
  • mto: (2520.5.2 bzr.mpbundle)
  • mto: This revision was merged to the branch mainline in revision 2631.
  • Revision ID: abentley@panoramicfeedback.com-20070621234317-5w3h8h36oe90sups
Implement new merge directive format

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/benchmarks/__init__.py
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Benchmark test suite for bzr."""
18
18
 
19
19
from bzrlib import (
20
20
    bzrdir,
 
21
    plugin,
21
22
    )
22
 
from bzrlib import plugin as _mod_plugin
23
23
import bzrlib.branch
24
24
from bzrlib.tests.TestUtil import TestLoader
25
25
from bzrlib.tests.blackbox import ExternalBase
26
26
 
27
27
 
28
28
class Benchmark(ExternalBase):
29
 
    """A Test class which provides helpers for writing benchmark tests."""
30
29
 
31
30
    def make_kernel_like_tree(self, url=None, root='.',
32
31
                              link_working=False):
33
32
        """Setup a temporary tree roughly like a kernel tree.
34
 
 
 
33
        
35
34
        :param url: Creat the kernel like tree as a lightweight checkout
36
 
            of a new branch created at url.
37
 
        :param root: Path where the tree will be created.
 
35
        of a new branch created at url.
38
36
        :param link_working: instead of creating a new copy of all files
39
37
            just hardlink the working tree. Tests must request this, because
40
38
            they must break links if they want to change the files
41
 
        :return: A newly created tree.
42
39
        """
43
40
        from bzrlib.benchmarks.tree_creator.kernel_like import (
44
41
            KernelLikeTreeCreator,
77
74
        :param link_working: Instead of copying all of the working tree
78
75
            files, just hardlink them to the cached files. Tests can unlink
79
76
            files that they will change.
80
 
        :param link_bzr: Hardlink the .bzr directory. For readonly
 
77
        :param link_bzr: Hardlink the .bzr directory. For readonly 
81
78
            operations this is safe, and shaves off a lot of setup time
82
79
        """
83
80
        from bzrlib.benchmarks.tree_creator.kernel_like import (
104
101
    def make_many_commit_tree(self, directory_name='.',
105
102
                              hardlink=False):
106
103
        """Create a tree with many commits.
107
 
 
108
 
        No file changes are included. Not hardlinking the working tree,
 
104
        
 
105
        No file changes are included. Not hardlinking the working tree, 
109
106
        because there are no working tree files.
110
107
        """
111
108
        from bzrlib.benchmarks.tree_creator.simple_many_commit import (
117
114
    def make_heavily_merged_tree(self, directory_name='.',
118
115
                                 hardlink=False):
119
116
        """Create a tree in which almost every commit is a merge.
120
 
 
121
 
        No file changes are included.  This produces two trees,
 
117
       
 
118
        No file changes are included.  This produces two trees, 
122
119
        one of which is returned.  Except for the first commit, every
123
120
        commit in its revision-history is a merge another commit in the other
124
 
        tree.  Not hardlinking the working tree, because there are no working
 
121
        tree.  Not hardlinking the working tree, because there are no working 
125
122
        tree files.
126
123
        """
127
124
        from bzrlib.benchmarks.tree_creator.heavily_merged import (
134
131
                            hardlink=False):
135
132
        """Create a tree with many files and many commits. Every commit changes
136
133
        exactly one file.
137
 
 
 
134
        
138
135
        :param num_files: number of files to be created
139
136
        :param num_commits: number of commits in the newly created tree
140
137
        """
156
153
        :param tree: The tree in which the changes happen.
157
154
        :param files: The list of files where changes should occur.
158
155
        :param num_commits: The number of commits
159
 
        :param changes_per_commit: The number of files that are touched in
160
 
            each commit.
 
156
        :param changes_per_commit: The number of files that are touched in 
 
157
        each commit.
161
158
        """
162
159
        for j in range(num_commits):
163
160
            for i in range(changes_per_commit):
180
177
                   'bzrlib.benchmarks.bench_cache_utf8',
181
178
                   'bzrlib.benchmarks.bench_checkout',
182
179
                   'bzrlib.benchmarks.bench_commit',
183
 
                   'bzrlib.benchmarks.bench_dirstate',
184
180
                   'bzrlib.benchmarks.bench_info',
185
181
                   'bzrlib.benchmarks.bench_inventory',
186
 
                   'bzrlib.benchmarks.bench_knit',
187
182
                   'bzrlib.benchmarks.bench_log',
188
 
                   'bzrlib.benchmarks.bench_pack',
189
183
                   'bzrlib.benchmarks.bench_osutils',
190
184
                   'bzrlib.benchmarks.bench_rocks',
191
185
                   'bzrlib.benchmarks.bench_startup',
192
186
                   'bzrlib.benchmarks.bench_status',
193
 
                   'bzrlib.benchmarks.bench_tags',
194
187
                   'bzrlib.benchmarks.bench_transform',
195
188
                   'bzrlib.benchmarks.bench_workingtree',
196
189
                   'bzrlib.benchmarks.bench_sftp',
197
190
                   'bzrlib.benchmarks.bench_xml',
198
191
                   ]
199
 
    suite = TestLoader().loadTestsFromModuleNames(testmod_names)
 
192
    suite = TestLoader().loadTestsFromModuleNames(testmod_names) 
200
193
 
201
194
    # Load any benchmarks from plugins
202
 
    for name, plugin in _mod_plugin.plugins().items():
203
 
        if getattr(plugin.module, 'bench_suite', None) is not None:
204
 
            suite.addTest(plugin.module.bench_suite())
 
195
    for name, module in plugin.all_plugins().items():
 
196
        if getattr(module, 'bench_suite', None) is not None:
 
197
            suite.addTest(module.bench_suite())
205
198
 
206
199
    return suite