[GR-17350] Add new specialization to isSuperSet node

PullRequest: graalpython/596

Author: Franziska Geiger

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

@@ -187,6 +187,15 @@ def test_sub_and_super():
         assert not set('cbs').issuperset('a')
 
 
+def test_superset_list():
+    set = {1, 2, 3, 4}
+    list = [1, 2, 3, 4]
+    visited= False
+    if set.issuperset(list):
+        visited = True
+    assert visited
+
+
 def test_intersection():
     word = 'simsalabim'
     otherword = 'madagascar'

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/set/FrozenSetBuiltins.java

@@ -72,6 +72,7 @@
 import com.oracle.graal.python.runtime.PythonContext;
 import com.oracle.graal.python.runtime.exception.PException;
 import com.oracle.graal.python.runtime.exception.PythonErrorType;
+import com.oracle.graal.python.runtime.sequence.PSequence;
 import com.oracle.truffle.api.CompilerDirectives;
 import com.oracle.truffle.api.CompilerDirectives.CompilationFinal;
 import com.oracle.truffle.api.dsl.Cached;
@@ -489,7 +490,15 @@ boolean isSuperSet(VirtualFrame frame, PBaseSet self, PBaseSet other,
         }
 
         @Specialization
-        boolean isSuperSet(VirtualFrame frame, PBaseSet self, String other,
+        boolean isSuperSetPSequence(VirtualFrame frame, PBaseSet self, PSequence other,
+                        @Cached("create()") SetNodes.ConstructSetNode constructSetNode,
+                        @Cached("create()") HashingStorageNodes.KeysIsSupersetNode isSupersetNode) {
+            PSet otherSet = constructSetNode.executeWith(frame, other);
+            return isSupersetNode.execute(frame, self.getDictStorage(), otherSet.getDictStorage());
+        }
+
+        @Specialization
+        boolean isSuperSetString(VirtualFrame frame, PBaseSet self, String other,
                         @Cached("create()") SetNodes.ConstructSetNode constructSetNode,
                         @Cached("create()") HashingStorageNodes.KeysIsSupersetNode isSupersetNode) {
             PSet otherSet = constructSetNode.executeWith(frame, other);