Adds groupby test with lambda