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