def test_remove(self):
heap = RemovalHeap(reversed(ascii_uppercase))
for c in reversed(ascii_uppercase):
wanted = set(heap)
wanted.remove(c)
heap.remove(c)
heap.check()
self.assertSetEqual(wanted, set(heap))
self.assertSetEqual(set(), set(heap))
def filled_heap(self):
heap = RemovalHeap(digits + ascii_uppercase)
for c in digits:
heap.remove(c)
return heap
def filled_heap(self):
heap = RemovalHeap(digits + ascii_uppercase)
for c in digits:
heap.remove(c)
return heap
