Alle Dateien aus dem Pythonkurs
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

26 lines
659 B

def get_groupby_method_args(name, obj):
"""
Get required arguments for a groupby method.
When parametrizing a test over groupby methods (e.g. "sum", "mean", "fillna"),
it is often the case that arguments are required for certain methods.
Parameters
----------
name: str
Name of the method.
obj: Series or DataFrame
pandas object that is being grouped.
Returns
-------
A tuple of required arguments for the method.
"""
if name in ("nth", "fillna", "take"):
return (0,)
if name == "quantile":
return (0.5,)
if name == "corrwith":
return (obj,)
return ()