/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/test__simple_set.py

  • Committer: Jelmer Vernooij
  • Date: 2017-06-10 01:35:53 UTC
  • mto: (6670.4.8 move-bzr)
  • mto: This revision was merged to the branch mainline in revision 6681.
  • Revision ID: jelmer@jelmer.uk-20170610013553-560y7mn3su4pp763
Fix remaining tests.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test__simple_set.py
1
 
# Copyright (C) 2009, 2010 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
18
18
 
19
19
import sys
20
20
 
21
 
from bzrlib import (
22
 
    errors,
23
 
    osutils,
 
21
from breezy import (
24
22
    tests,
25
23
    )
 
24
from breezy.tests import (
 
25
    features,
 
26
    )
26
27
 
27
28
try:
28
 
    from bzrlib import _simple_set_pyx
 
29
    from breezy import _simple_set_pyx
29
30
except ImportError:
30
31
    _simple_set_pyx = None
31
32
 
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
 
compiled_simpleset_feature = tests.ModuleAvailableFeature(
81
 
                                'bzrlib._simple_set_pyx')
 
81
compiled_simpleset_feature = features.ModuleAvailableFeature(
 
82
                                'breezy._simple_set_pyx')
82
83
 
83
84
 
84
85
class TestSimpleSet(tests.TestCase):
372
373
            all.add(key)
373
374
        self.assertEqual(sorted([k1, k2, k3]), sorted(all))
374
375
        iterator = iter(obj)
375
 
        iterator.next()
 
376
        next(iterator)
376
377
        obj.add(('foo',))
377
378
        # Set changed size
378
 
        self.assertRaises(RuntimeError, iterator.next)
 
379
        self.assertRaises(RuntimeError, next, iterator)
379
380
        # And even removing an item still causes it to fail
380
381
        obj.discard(k2)
381
 
        self.assertRaises(RuntimeError, iterator.next)
 
382
        self.assertRaises(RuntimeError, next, iterator)
 
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)