TST: Updating test of groupby