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