我有类似的代码:
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)。但这感觉就像在作弊...