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