← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test__simple_set.py

(jelmer) Use the absolute_import feature everywhere in bzrlib,
 and add a source test to make sure it's used everywhere. (Jelmer Vernooij)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test__simple_set.py
1
 
# Copyright (C) 2009 Canonical Ltd
 
1
# Copyright (C) 2009, 2010, 2011 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
19
19
import sys
20
20
 
21
21
from bzrlib import (
22
 
    errors,
23
 
    osutils,
24
22
    tests,
25
23
    )
 
24
from bzrlib.tests import (
 
25
    features,
 
26
    )
26
27
 
27
28
try:
28
29
    from bzrlib import _simple_set_pyx
77
78
 
78
79
# Even though this is an extension, we don't permute the tests for a python
79
80
# version. As the plain python version is just a dict or set
80
 
 
81
 
class _CompiledSimpleSet(tests.Feature):
82
 
 
83
 
    def _probe(self):
84
 
        if _simple_set_pyx is None:
85
 
            return False
86
 
        return True
87
 
 
88
 
    def feature_name(self):
89
 
        return 'bzrlib._simple_set_pyx'
90
 
 
91
 
CompiledSimpleSet = _CompiledSimpleSet()
 
81
compiled_simpleset_feature = features.ModuleAvailableFeature(
 
82
                                'bzrlib._simple_set_pyx')
92
83
 
93
84
 
94
85
class TestSimpleSet(tests.TestCase):
95
86
 
96
 
    _test_needs_features = [CompiledSimpleSet]
 
87
    _test_needs_features = [compiled_simpleset_feature]
97
88
    module = _simple_set_pyx
98
89
 
99
90
    def assertIn(self, obj, container):
389
380
        # And even removing an item still causes it to fail
390
381
        obj.discard(k2)
391
382
        self.assertRaises(RuntimeError, iterator.next)
 
383
 
 
384
    def test__sizeof__(self):
 
385
        # SimpleSet needs a custom sizeof implementation, because it allocates
 
386
        # memory that Python cannot directly see (_table).
 
387
        # Too much variability in platform sizes for us to give a fixed size
 
388
        # here. However without a custom implementation, __sizeof__ would give
 
389
        # us only the size of the object, and not its table. We know the table
 
390
        # is at least 4bytes*1024entries in size.
 
391
        obj = self.module.SimpleSet()
 
392
        self.assertTrue(obj.__sizeof__() > 4096)