TST: fix groupby_sort tests