SI-8264 scala.collection.immutable.HashSet#- returns broken Set

Was an error in HashSet's handling of removal of an element when a HashTrieSet should turn into a HashSet1.

Also slightly modified HashMap's filter0 to more closely match HashSet (by adding the same comment).

1 files changed, 5 insertions, 6 deletions

diff --git a/src/library/scala/collection/immutable/HashSet.scala b/src/library/scala/collection/immutable/HashSet.scala
index 67e9d18da7..726937efd9 100644
--- a/src/library/scala/collection/immutable/HashSet.scala
+++ b/src/library/scala/collection/immutable/HashSet.scala
@@ -431,15 +431,12 @@ object HashSet extends ImmutableSetFactory[HashSet] {
           case 0 =>
             // the empty set
             null
-          case size if size == ks.size =>
-            // We do this check first since even if the result is of size 1 since
-            // it is preferable to return the existing set for better structural sharing
-            // we can not rely on ks.- returning the same instance if we subtract an element that is not in it
-            // so we need to do the size check
-            this
           case 1 =>
             // create a new HashSet1 with the hash we already know
             new HashSet1(ks1.head, hash)
+          case size if size == ks.size =>
+            // Should only have HSC1 if size > 1
+            this
           case _ =>
             // create a new HashSetCollison with the hash we already know and the new keys
             new HashSetCollision1(hash, ks1)
@@ -861,6 +858,8 @@ object HashSet extends ImmutableSetFactory[HashSet] {
               new HashTrieSet(bitmapNew, elemsNew, sizeNew)
           } else
             null
+        } else if(elems.length == 1 && !subNew.isInstanceOf[HashTrieSet[_]]) {
+          subNew
         } else {
           val elemsNew = new Array[HashSet[A]](elems.length)
           Array.copy(elems, 0, elemsNew, 0, elems.length)