예제 #1
0
def test_series_nsmallest(data, n):
    """Indirectly tests Series.sort_values()
    """
    sr = Series(data)
    psr = pd.Series(data)
    assert_eq(sr.nsmallest(n), psr.nsmallest(n))
    assert_eq(
        sr.nsmallest(n, keep="last").sort_index(),
        psr.nsmallest(n, keep="last").sort_index(),
    )

    assert_exceptions_equal(
        lfunc=psr.nsmallest,
        rfunc=sr.nsmallest,
        lfunc_args_and_kwargs=([], {"n": 3, "keep": "what"}),
        rfunc_args_and_kwargs=([], {"n": 3, "keep": "what"}),
        expected_error_message='keep must be either "first", "last"',
    )