Ejemplo n.º 1
0
def test_series_argsort(nelem, dtype, asc):
    np.random.seed(0)
    sr = Series((100 * np.random.random(nelem)).astype(dtype))
    res = sr.argsort(ascending=asc)

    if asc:
        expected = np.argsort(sr.to_array(), kind="mergesort")
    else:
        expected = np.argsort(sr.to_array() * -1, kind="mergesort")
    np.testing.assert_array_equal(expected, res.to_array())