TST: Adding tests for generic_groupby enhancement