~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test__chk_map.py

  • Committer: Andrew Bennetts
  • Date: 2010-01-15 03:58:20 UTC
  • mfrom: (4963 +trunk)
  • mto: (4973.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 4975.
  • Revision ID: andrew.bennetts@canonical.com-20100115035820-ilb3t36swgzq6v1l
MergeĀ lp:bzr.

Show diffs side-by-side

added added

removed removed

Lines of Context:
25
25
 
26
26
 
27
27
def load_tests(standard_tests, module, loader):
28
 
    # parameterize all tests in this module
29
 
    suite = loader.suiteClass()
30
 
    import bzrlib._chk_map_py as py_module
31
 
    scenarios = [('python', {'module': py_module})]
32
 
    if CompiledChkMapFeature.available():
33
 
        import bzrlib._chk_map_pyx as c_module
34
 
        scenarios.append(('C', {'module': c_module}))
35
 
    else:
36
 
        # the compiled module isn't available, so we add a failing test
37
 
        class FailWithoutFeature(tests.TestCase):
38
 
            def test_fail(self):
39
 
                self.requireFeature(CompiledChkMapFeature)
40
 
        suite.addTest(loader.loadTestsFromTestCase(FailWithoutFeature))
41
 
    tests.multiply_tests(standard_tests, scenarios, suite)
 
28
    suite, _ = tests.permute_tests_for_extension(standard_tests, loader,
 
29
        'bzrlib._chk_map_py', 'bzrlib._chk_map_pyx')
42
30
    return suite
43
31
 
44
32
 
45
 
class _CompiledChkMapFeature(tests.Feature):
46
 
 
47
 
    def _probe(self):
48
 
        try:
49
 
            import bzrlib._chk_map_pyx
50
 
        except ImportError:
51
 
            return False
52
 
        return True
53
 
 
54
 
    def feature_name(self):
55
 
        return 'bzrlib._chk_map_pyx'
56
 
 
57
 
CompiledChkMapFeature = _CompiledChkMapFeature()
58
 
 
59
 
 
60
33
class TestSearchKeys(tests.TestCase):
61
34
 
62
35
    module = None # Filled in by test parameterization