Add a thre column test for groupby sum