add basic support for issubset issupperset set/frozenset builtins

Author: cosminbasca

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

@@ -60,6 +60,7 @@
 import com.oracle.graal.python.builtins.objects.common.HashingStorageNodesFactory.GetItemNodeGen;
 import com.oracle.graal.python.builtins.objects.common.HashingStorageNodesFactory.InitNodeGen;
 import com.oracle.graal.python.builtins.objects.common.HashingStorageNodesFactory.KeysEqualsNodeGen;
+import com.oracle.graal.python.builtins.objects.common.HashingStorageNodesFactory.KeysIsSubsetNodeGen;
 import com.oracle.graal.python.builtins.objects.common.HashingStorageNodesFactory.SetItemNodeGen;
 import com.oracle.graal.python.builtins.objects.common.HashingStorageNodesFactory.UnionNodeGen;
 import com.oracle.graal.python.builtins.objects.dict.PDict;
@@ -1386,6 +1387,48 @@ public static ExclusiveOrNode create() {
         }
     }
 
+    public abstract static class KeysIsSubsetNode extends DictStorageBaseNode {
+
+        public abstract boolean execute(HashingStorage left, HashingStorage right);
+
+        @Specialization
+        public boolean isSubset(HashingStorage left, HashingStorage right,
+                        @Cached("create()") ContainsKeyNode containsKeyNode,
+                        @Cached("createBinaryProfile()") ConditionProfile sizeProfile) {
+            if (sizeProfile.profile(left.length() > right.length())) {
+                return false;
+            }
+
+            for (Object leftKey : left.keys()) {
+                if (!containsKeyNode.execute(right, leftKey)) {
+                    return false;
+                }
+            }
+            return true;
+        }
+
+        public static KeysIsSubsetNode create() {
+            return KeysIsSubsetNodeGen.create();
+        }
+    }
+
+    public static class KeysIsSupersetNode extends Node {
+        @Child KeysIsSubsetNode isSubsetNode;
+
+        public boolean execute(HashingStorage left, HashingStorage right) {
+            if (isSubsetNode == null) {
+                CompilerDirectives.transferToInterpreterAndInvalidate();
+                isSubsetNode = insert(KeysIsSubsetNode.create());
+            }
+
+            return isSubsetNode.execute(right, left);
+        }
+
+        public static KeysIsSupersetNode create() {
+            return new KeysIsSupersetNode();
+        }
+    }
+
     public abstract static class DiffNode extends DictStorageBaseNode {
 
         public abstract HashingStorage execute(HashingStorage left, HashingStorage right);

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

@@ -287,6 +287,41 @@ PBaseSet doIterable(PBaseSet container, HashingStorage dictStorage, Object itera
         public static BinaryUnionNode create() {
             return BinaryUnionNodeGen.create();
         }
+    }
+
+    @Builtin(name = "issubset", fixedNumOfArguments = 2)
+    @GenerateNodeFactory
+    abstract static class IsSubsetNode extends PythonBinaryBuiltinNode {
+        @Specialization
+        boolean isSubSet(PBaseSet self, PBaseSet other,
+                        @Cached("create()") HashingStorageNodes.KeysIsSubsetNode isSubsetNode) {
+            return isSubsetNode.execute(self.getDictStorage(), other.getDictStorage());
+        }
 
+        @Specialization
+        boolean isSubSet(PBaseSet self, String other,
+                        @Cached("create()") SetNodes.ConstructSetNode constructSetNode,
+                        @Cached("create()") HashingStorageNodes.KeysIsSubsetNode isSubsetNode) {
+            PSet otherSet = constructSetNode.executeWith(other);
+            return isSubsetNode.execute(self.getDictStorage(), otherSet.getDictStorage());
+        }
+    }
+
+    @Builtin(name = "issuperset", fixedNumOfArguments = 2)
+    @GenerateNodeFactory
+    abstract static class IsSupersetNode extends PythonBinaryBuiltinNode {
+        @Specialization
+        boolean isSuperSet(PBaseSet self, PBaseSet other,
+                        @Cached("create()") HashingStorageNodes.KeysIsSupersetNode isSupersetNode) {
+            return isSupersetNode.execute(self.getDictStorage(), other.getDictStorage());
+        }
+
+        @Specialization
+        boolean isSuperSet(PBaseSet self, String other,
+                        @Cached("create()") SetNodes.ConstructSetNode constructSetNode,
+                        @Cached("create()") HashingStorageNodes.KeysIsSupersetNode isSupersetNode) {
+            PSet otherSet = constructSetNode.executeWith(other);
+            return isSupersetNode.execute(self.getDictStorage(), otherSet.getDictStorage());
+        }
     }
 }