add basic support for < > <= >= set and frozenset ops

Author: cosminbasca

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

@@ -28,9 +28,12 @@
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__AND__;
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__CONTAINS__;
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__EQ__;
+import static com.oracle.graal.python.nodes.SpecialMethodNames.__GE__;
+import static com.oracle.graal.python.nodes.SpecialMethodNames.__GT__;
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__ITER__;
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__LEN__;
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__LE__;
+import static com.oracle.graal.python.nodes.SpecialMethodNames.__LT__;
 import static com.oracle.graal.python.nodes.SpecialMethodNames.__SUB__;
 
 import java.util.List;
@@ -324,4 +327,78 @@ boolean isSuperSet(PBaseSet self, String other,
             return isSupersetNode.execute(self.getDictStorage(), otherSet.getDictStorage());
         }
     }
+
+    @Builtin(name = __LE__, fixedNumOfArguments = 2)
+    @GenerateNodeFactory
+    abstract static class LessEqualNode extends IsSubsetNode {
+    }
+
+    @Builtin(name = __GE__, fixedNumOfArguments = 2)
+    @GenerateNodeFactory
+    abstract static class GreaterEqualNode extends IsSupersetNode {
+    }
+
+    @Builtin(name = __LT__, fixedNumOfArguments = 2)
+    @GenerateNodeFactory
+    abstract static class LessThanNode extends PythonBinaryBuiltinNode {
+        @Child LessEqualNode lessEqualNode;
+
+        private LessEqualNode getLessEqualNode() {
+            if (lessEqualNode == null) {
+                CompilerDirectives.transferToInterpreterAndInvalidate();
+                lessEqualNode = insert(FrozenSetBuiltinsFactory.LessEqualNodeFactory.create());
+            }
+            return lessEqualNode;
+        }
+
+        @Specialization
+        boolean isLessThan(PBaseSet self, PBaseSet other,
+                        @Cached("createBinaryProfile()") ConditionProfile sizeProfile) {
+            if (sizeProfile.profile(self.size() >= other.size())) {
+                return false;
+            }
+            return (Boolean) getLessEqualNode().execute(self, other);
+        }
+
+        @Specialization
+        boolean isLessThan(PBaseSet self, String other,
+                        @Cached("createBinaryProfile()") ConditionProfile sizeProfile) {
+            if (sizeProfile.profile(self.size() >= other.length())) {
+                return false;
+            }
+            return (Boolean) getLessEqualNode().execute(self, other);
+        }
+    }
+
+    @Builtin(name = __GT__, fixedNumOfArguments = 2)
+    @GenerateNodeFactory
+    abstract static class GreaterThanNode extends PythonBinaryBuiltinNode {
+        @Child GreaterEqualNode greaterEqualNode;
+
+        private GreaterEqualNode getGreaterEqualNode() {
+            if (greaterEqualNode == null) {
+                CompilerDirectives.transferToInterpreterAndInvalidate();
+                greaterEqualNode = insert(FrozenSetBuiltinsFactory.GreaterEqualNodeFactory.create());
+            }
+            return greaterEqualNode;
+        }
+
+        @Specialization
+        boolean isGreaterThan(PBaseSet self, PBaseSet other,
+                        @Cached("createBinaryProfile()") ConditionProfile sizeProfile) {
+            if (sizeProfile.profile(self.size() <= other.size())) {
+                return false;
+            }
+            return (Boolean) getGreaterEqualNode().execute(self, other);
+        }
+
+        @Specialization
+        boolean isGreaterThan(PBaseSet self, String other,
+                        @Cached("createBinaryProfile()") ConditionProfile sizeProfile) {
+            if (sizeProfile.profile(self.size() <= other.length())) {
+                return false;
+            }
+            return (Boolean) getGreaterEqualNode().execute(self, other);
+        }
+    }
 }