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