← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/benchmarks/bench_knit.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-01-26 10:49:57 UTC
  • mfrom: (4987.1.1 integration)
  • Revision ID: pqm@pqm.ubuntu.com-20100126104957-dmtqnc0pckuruyla
(vila,
        jam) Implement TestCase.overrideAttr to simplify tests setUp/cleanup

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/benchmarks/bench_knit.py
61
61
 
62
62
    def setup_load_data_c(self):
63
63
        self.requireFeature(CompiledKnitFeature)
64
 
        orig = knit._load_data
65
 
        def reset():
66
 
            knit._load_data = orig
67
 
        self.addCleanup(reset)
68
64
        from bzrlib._knit_load_data_pyx import _load_data_c
69
 
        knit._load_data = _load_data_c
 
65
        self.overrideAttr(knit, '_load_data', _load_data_c)
70
66
 
71
67
    def setup_load_data_py(self):
72
 
        orig = knit._load_data
73
 
        def reset():
74
 
            knit._load_data = orig
75
 
        self.addCleanup(reset)
76
68
        from bzrlib._knit_load_data_py import _load_data_py
77
 
        knit._load_data = _load_data_py
 
69
        self.overrideAttr(knit, '_load_data', _load_data_py)
78
70
 
79
71
    def test_read_50k_index_c(self):
80
72
        self.setup_load_data_c()