Added tests in SortedEquaSetSpec to ensure TreeEquaSet groupBy method return type uses TreeEquaSet.