[GR-10614] Missing implementation of special functions from Math module.

PullRequest: graalpython/99

Author: ppisl

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

@@ -0,0 +1,27 @@
+# Copyright (c) 2018, Oracle and/or its affiliates.
+# Copyright (C) 1996-2017 Python Software Foundation
+#
+# Licensed under the PYTHON SOFTWARE FOUNDATION LICENSE VERSION 2
+    
+import unittest
+
+class MyIndexable(object):
+    def __init__(self, value):
+        self.value = value
+    def __index__(self):
+        return self.value
+
+class BuiltinTest(unittest.TestCase):
+    def test_bin(self):
+        self.assertEqual(bin(0), '0b0')
+        self.assertEqual(bin(1), '0b1')
+        self.assertEqual(bin(-1), '-0b1')
+        self.assertEqual(bin(2**65), '0b1' + '0' * 65)
+        self.assertEqual(bin(2**65-1), '0b' + '1' * 65)
+        self.assertEqual(bin(-(2**65)), '-0b1' + '0' * 65)
+        self.assertEqual(bin(-(2**65-1)), '-0b' + '1' * 65)
+
+        # test of specializations
+        self.assertEqual(bin(MyIndexable(False)), '0b0')
+        self.assertEqual(bin(MyIndexable(True)), '0b1')
+        self.assertEqual(bin(MyIndexable(-(2**65))), '-0b1' + '0' * 65)

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

@@ -28,6 +28,7 @@
 import static com.oracle.graal.python.builtins.objects.PNone.NO_VALUE;
 import static com.oracle.graal.python.builtins.objects.PNotImplemented.NOT_IMPLEMENTED;
 import static com.oracle.graal.python.nodes.BuiltinNames.ABS;
+import static com.oracle.graal.python.nodes.BuiltinNames.BIN;
 import static com.oracle.graal.python.nodes.BuiltinNames.CALLABLE;
 import static com.oracle.graal.python.nodes.BuiltinNames.CHR;
 import static com.oracle.graal.python.nodes.BuiltinNames.COMPILE;
@@ -120,6 +121,7 @@
 import com.oracle.graal.python.nodes.object.GetClassNode;
 import com.oracle.graal.python.nodes.subscript.GetItemNode;
 import com.oracle.graal.python.nodes.truffle.PythonArithmeticTypes;
+import com.oracle.graal.python.nodes.util.CastToIntNode;
 import com.oracle.graal.python.runtime.PythonContext;
 import com.oracle.graal.python.runtime.PythonCore;
 import com.oracle.graal.python.runtime.PythonOptions;
@@ -198,6 +200,54 @@ public Object absObject(Object object,
         }
     }
 
+    // bin(object)
+    @Builtin(name = BIN, fixedNumOfArguments = 1)
+    @TypeSystemReference(PythonArithmeticTypes.class)
+    @GenerateNodeFactory
+    public abstract static class BinNode extends PythonUnaryBuiltinNode {
+
+        public abstract String executeObject(Object x);
+
+        private static String buildString(boolean isNegative, String number) {
+            StringBuilder sb = new StringBuilder();
+            if (isNegative) {
+                sb.append('-');
+            }
+            sb.append("0b");
+            sb.append(number);
+            return sb.toString();
+        }
+
+        @Specialization
+        public String doL(long x) {
+            return buildString(x < 0, Long.toBinaryString(Math.abs(x)));
+        }
+
+        @Specialization
+        public String doD(double x) {
+            throw raise(TypeError, "'%p' object cannot be interpreted as an integer", x);
+        }
+
+        @Specialization
+        @TruffleBoundary
+        public String doPI(PInt x) {
+            BigInteger value = x.getValue();
+            return buildString(value.compareTo(BigInteger.ZERO) == -1, value.abs().toString(2));
+        }
+
+        @Specialization
+        public String doO(Object x,
+                        @Cached("create()") CastToIntNode toIntNode,
+                        @Cached("create()") BinNode recursiveNode) {
+            Object value = toIntNode.execute(x);
+            return recursiveNode.executeObject(value);
+        }
+
+        protected BinNode create() {
+            return BuiltinFunctionsFactory.BinNodeFactory.create();
+        }
+    }
+
     // callable(object)
     @Builtin(name = CALLABLE, fixedNumOfArguments = 1)
     @GenerateNodeFactory