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"', )