Fixed test case for heap.py