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