熊猫的UnitTest pandas.core.groupby.GroupBy.apply方法

时间:2019-06-25 10:49:38

标签: python pandas python-unittest

我有类似的代码:

def bla (sdf, blub):
    (a, b) = sdf.name
    # do stuff, using a and b
    return pd.DataFrame()

df.groupby(["foo", "bar"]).apply(bla, blub=blub)

我想对bla方法进行单元测试。 (多余的blub参数不应干扰...)

所以我的问题是:

如何模拟DataFrameGroupBy对象?

我需要将name属性设置为tupel(a,b)并调用

一个简单的

   foo = DataFrameGroupBy(bla)

没用...

编辑1: 解决方法(w | c | sh)应使用仅一行的数据帧df1,并使用df1.groupyby(....)。apply(bla,blub)。但这感觉就像在作弊...

0 个答案:

没有答案