enhanced groupby test