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