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)