~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/intset.py

  • Committer: Aaron Bentley
  • Date: 2005-09-19 01:19:59 UTC
  • mto: (1185.1.29)
  • mto: This revision was merged to the branch mainline in revision 1390.
  • Revision ID: aaron.bentley@utoronto.ca-20050919011959-9cf31f97b991b08d
Switched NULL revision ID to 'null:' per robertc's suggestion

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
#! /usr/bin/python
 
2
 
1
3
# Copyright (C) 2005 Canonical Ltd
2
4
 
3
5
# This program is free software; you can redistribute it and/or modify
208
210
        if not self._val & m:
209
211
            raise KeyError(to_remove)
210
212
        self._val ^= m
211
 
 
212
 
    def set_remove(self, to_remove):
213
 
        """Remove all values that exist in to_remove.
214
 
 
215
 
        >>> a = IntSet(range(10))
216
 
        >>> b = IntSet([2,3,4,7,12])
217
 
        >>> a.set_remove(b)
218
 
        >>> list(a)
219
 
        [0, 1, 5, 6, 8, 9]
220
 
        >>> a.set_remove([1,2,5])
221
 
        >>> list(a)
222
 
        [0, 6, 8, 9]
223
 
        """
224
 
        if not isinstance(to_remove, IntSet):
225
 
            self.set_remove(IntSet(to_remove))
226
 
            return
227
 
        intersect = self._val & to_remove._val
228
 
        self._val ^= intersect
229
 
 
 
213
        
 
214
        
 
215
            
 
216
    
 
217
 
 
218
if __name__ == '__main__':
 
219
    import doctest
 
220
    doctest.testmod()
 
221