[GR-15758] missing specialization for set delete

PullRequest: graalpython/513

Author: Franziska Geiger

graalpython/com.oracle.graal.python.test/src/tests/test_set.py

@@ -272,3 +272,15 @@ def test_pop():
     except BaseException as e:
         assert type(e) == KeyError, "expected KeyError, got %s" % type(e)
 
+
+def test_set_delete():
+    s = {1, 2, 3}
+    assert s == {1, 2, 3}
+    s.discard(3)
+    assert s == {1, 2}
+
+    # string keys
+    s = {'a', 'b', 'c'}
+    assert s == {'a', 'b', 'c'}
+    s.discard('c')
+    assert s == {'a', 'b'}

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/common/HashingStorageNodes.java

@@ -75,6 +75,7 @@
 import com.oracle.graal.python.builtins.objects.function.PKeyword;
 import com.oracle.graal.python.builtins.objects.ints.PInt;
 import com.oracle.graal.python.builtins.objects.object.PythonObject;
+import com.oracle.graal.python.builtins.objects.set.PSet;
 import com.oracle.graal.python.builtins.objects.str.LazyString;
 import com.oracle.graal.python.builtins.objects.str.PString;
 import com.oracle.graal.python.builtins.objects.type.LazyPythonClass;
@@ -1496,11 +1497,21 @@ protected boolean doDynamicObjectString(@SuppressWarnings("unused") PDict contai
             return storage.remove(key, DEFAULT_EQIVALENCE);
         }
 
+        @Specialization
+        protected boolean doDynamicObjectString(@SuppressWarnings("unused") PSet container, DynamicObjectStorage storage, String key) {
+            return storage.remove(key, DEFAULT_EQIVALENCE);
+        }
+
         @Specialization(guards = "wrappedString(key)")
         protected boolean doDynamicObjectPString(@SuppressWarnings("unused") PDict container, DynamicObjectStorage storage, PString key) {
             return storage.remove(key.getValue(), DEFAULT_EQIVALENCE);
         }
 
+        @Specialization(guards = "wrappedString(key)")
+        protected boolean doDynamicObjectPString(@SuppressWarnings("unused") PSet container, DynamicObjectStorage storage, PString key) {
+            return storage.remove(key.getValue(), DEFAULT_EQIVALENCE);
+        }
+
         @SuppressWarnings("unused")
         @Specialization(guards = "!isJavaString(key)")
         protected boolean doDynamicObject(PHashingCollection container, DynamicObjectStorage storage, Object key) {