← 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

  • Committer: Robert Collins
  • Date: 2009-12-16 22:29:31 UTC
  • mto: This revision was merged to the branch mainline in revision 4920.
  • Revision ID: robertc@robertcollins.net-20091216222931-wbbn5ey4mwmpatwd
Review feedback.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test__simple_set.py
1
 
# Copyright (C) 2009, 2010, 2011 Canonical Ltd
 
1
# Copyright (C) 2009 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,
22
24
    tests,
23
25
    )
24
 
from bzrlib.tests import (
25
 
    features,
26
 
    )
27
26
 
28
27
try:
29
28
    from bzrlib import _simple_set_pyx
78
77
 
79
78
# Even though this is an extension, we don't permute the tests for a python
80
79
# version. As the plain python version is just a dict or set
81
 
compiled_simpleset_feature = features.ModuleAvailableFeature(
82
 
                                'bzrlib._simple_set_pyx')
 
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()
83
92
 
84
93
 
85
94
class TestSimpleSet(tests.TestCase):
86
95
 
87
 
    _test_needs_features = [compiled_simpleset_feature]
 
96
    _test_needs_features = [CompiledSimpleSet]
88
97
    module = _simple_set_pyx
89
98
 
90
99
    def assertIn(self, obj, container):
380
389
        # And even removing an item still causes it to fail
381
390
        obj.discard(k2)
382
391
        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)