Add tests for other versions of groupBy