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