← 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: Vincent Ladeuil
  • Date: 2010-01-26 10:17:22 UTC
  • mfrom: (4985.1.5 add-attr-cleanup)
  • mto: This revision was merged to the branch mainline in revision 4988.
  • Revision ID: v.ladeuil+lp@free.fr-20100126101722-y213sv39buw3dgi2
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()