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_numpy(), kind="mergesort")
    else:
        expected = np.argsort(sr.to_numpy() * -1, kind="mergesort")
    np.testing.assert_array_equal(expected, res.to_numpy())
Ejemplo n.º 2
0
def test_series_floor():
    arr = np.random.random(100) * 100
    sr = Series(arr)
    with pytest.warns(
        FutureWarning, match="Series.floor and DataFrame.floor are deprecated"
    ):
        sr = sr.floor()
    np.testing.assert_equal(sr.to_numpy(), np.floor(arr))