def test_all_repeat_heap(): lst = [2, 2, 2, 2, 2, 2] assert get_kleast_heap(lst, 0) == [] assert get_kleast_heap(lst, 1) == [2] assert get_kleast_heap(lst, 2) == [2, 2] assert get_kleast_heap(lst, 3) == [2, 2, 2] assert get_kleast_heap(lst, 7) == []
def test_part_repeat_heap(): lst = [4, 5, 1, 6, 2, 7, 2, 8] assert get_kleast_heap(lst, 0) == [] assert get_kleast_heap(lst, 1) == [1] assert get_kleast_heap(lst, 2) == [2, 1] assert get_kleast_heap(lst, 3) == [2, 2, 1] assert get_kleast_heap(lst, 5) == [5, 4, 2, 2, 1] assert get_kleast_heap(lst, 9) == []
def test_multi_repeat_heap(): lst = [2, 2, 1, 1, 3, 3] assert get_kleast_heap(lst, 0) == [] assert get_kleast_heap(lst, 1) == [1] assert get_kleast_heap(lst, 2) == [1, 1] assert get_kleast_heap(lst, 3) == [2, 1, 1] assert get_kleast_heap(lst, 5) == [3, 2, 2, 1, 1] assert get_kleast_heap(lst, 7) == []
def test_normal_heap(): lst = [4, 5, 1, 3, 2] assert get_kleast_heap(lst, 0) == [] assert get_kleast_heap(lst, 1) == [1] assert get_kleast_heap(lst, 2) == [2, 1] assert get_kleast_heap(lst, 3) == [3, 2, 1] assert get_kleast_heap(lst, 5) == [5, 4, 3, 2, 1] assert get_kleast_heap(lst, 6) == []
def test_one_heap(): lst = [1] assert get_kleast_heap(lst, 0) == [] assert get_kleast_heap(lst, 1) == [1] assert get_kleast_heap(lst, 2) == []