def test_heapsort():
rng = np.random.RandomState(0)
values = rng.rand(500)
indices = np.arange(len(values)).astype(np.int32)
sorted_idx = np.argsort(values)
sorted_values = values[sorted_idx]
sorted_indices = indices[sorted_idx]
heapsort(values, indices, len(values))
assert_array_equal(sorted_values, values)
assert_array_equal(sorted_indices, indices)
def test_heapsort_one():
values = np.arange(1).astype(np.float64)
indices = np.arange(1).astype(np.int32)
heapsort(values, indices, len(values))
