def test_dataframe_sort_values(nelem, dtype):
np.random.seed(0)
df = DataFrame()
df["a"] = aa = (100 * np.random.random(nelem)).astype(dtype)
df["b"] = bb = (100 * np.random.random(nelem)).astype(dtype)
sorted_df = df.sort_values(by="a")
# Check
sorted_index = np.argsort(aa, kind="mergesort")
assert_eq(sorted_df.index.values, sorted_index)
assert_eq(sorted_df["a"].values, aa[sorted_index])
assert_eq(sorted_df["b"].values, bb[sorted_index])
def test_dataframe_sort_values_ignore_index(index, ignore_index):
gdf = DataFrame({
"a": [1, 3, 5, 2, 4],
"b": [1, 1, 2, 2, 3],
"c": [9, 7, 7, 7, 1]
})
gdf = gdf.set_index(index)
pdf = gdf.to_pandas()
expect = pdf.sort_values(list(pdf.columns), ignore_index=ignore_index)
got = gdf.sort_values((gdf.columns), ignore_index=ignore_index)
assert_eq(expect, got)
