def test_series_nlargest(data, n):
"""Indirectly tests Series.sort_values()
"""
sr = Series(data)
psr = pd.Series(data)
assert_eq(sr.nlargest(n), psr.nlargest(n))
assert_eq(sr.nlargest(n, keep="last"), psr.nlargest(n, keep="last"))
with pytest.raises(ValueError) as raises:
sr.nlargest(3, keep="what")
assert raises.match('keep must be either "first", "last"')
def test_series_nlargest():
"""Indirectly tests Series.sort_values()
"""
sr = Series([0, 1, 1, 2, 2, 2, 3, 3])
got = sr.nlargest(3) # default keep='first'
assert list(got) == [3, 3, 2]
assert list(got.index.values) == [6, 7, 3]
got = sr.nlargest(3, keep="last")
assert list(got) == [3, 3, 2]
assert list(got.index.values) == [7, 6, 5]
with pytest.raises(ValueError) as raises:
sr.nlargest(3, keep="what")
assert raises.match('keep must be either "first", "last"')
def test_series_nlargest(data, n):
"""Indirectly tests Series.sort_values()
"""
sr = Series(data)
psr = pd.Series(data)
assert_eq(sr.nlargest(n), psr.nlargest(n))
assert_eq(sr.nlargest(n, keep="last"), psr.nlargest(n, keep="last"))
assert_exceptions_equal(
lfunc=psr.nlargest,
rfunc=sr.nlargest,
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"',
)
