def test_peek(self): hh = HashHeap(htype=MAX_HEAP) [hh.push(x, str(x)) for x in xrange(5)] assert hh.peek() == (4,'4') assert [x for x in hh.xpeek(count=2)] == [(4,'4'),(3,'3')]