add tests

Author: timfel

graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_abstract.py

@@ -1219,7 +1219,102 @@ def compile_module(self, name):
         arguments=["PyObject* haystack", "PyObject* needle"],
         cmpfunc=unhandled_error_compare
     )
-    
+
+    def index_func(args):
+        for idx,obj in enumerate(args[0]):
+            if obj == args[1]:
+                return idx
+        raise ValueError
+
+    test_PySequence_Index = CPyExtFunction(
+        index_func,
+        lambda: (
+            (tuple(), 1),
+            ((1, 2, 3), 1),
+            ((1, 2, 3), 4),
+            ((None,), 1),
+            ([], 1),
+            (['a', 'b', 'c'], 'a'),
+            (['a', 'b', 'c'], 'd'),
+            ([None], 1),
+            (set(), 1),
+            ({'a', 'b'}, 'a'),
+            ({'a', 'b'}, 'c'),
+            (DummyListSubclass(), 1),
+            ('hello', 'e'),
+            ('hello', 'x'),
+        ),
+        resultspec="i",
+        argspec='OO',
+        arguments=["PyObject* haystack", "PyObject* needle"],
+        cmpfunc=unhandled_error_compare
+    )
+
+    def count_func(args):
+        c = 0
+        for obj in args[0]:
+            if obj == args[1]:
+                c += 1
+        return c
+
+    test_PySequence_Count = CPyExtFunction(
+        count_func,
+        lambda: (
+            (tuple(), 1),
+            ((1, 2, 3), 1),
+            ((1, 2, 3), 4),
+            ((None,), 1),
+            ([], 1),
+            (['a', 'b', 'c'], 'a'),
+            (['a', 'b', 'c'], 'd'),
+            ([None], 1),
+            (set(), 1),
+            ({'a', 'b'}, 'a'),
+            ({'a', 'b'}, 'c'),
+            (DummyListSubclass(), 1),
+            ('hello', 'e'),
+            ('hello', 'x'),
+        ),
+        resultspec="i",
+        argspec='OO',
+        arguments=["PyObject* haystack", "PyObject* needle"],
+        cmpfunc=unhandled_error_compare
+    )
+
+    def _reference_setslice(args):
+        args[0][args[1]:args[2]] = args[3]
+        return 0;
+
+    test_PySequence_SetSlice = CPyExtFunction(
+        _reference_setslice,
+        lambda: (
+            ([1,2,3,4],0,4,[5,6,7,8]),
+            ([],1,2, [5,6]),
+            ([1,2,3,4],10,20,[5,6,7,8]),
+        ),
+        resultspec="i",
+        argspec='OnnO',
+        arguments=["PyObject* op", "Py_ssize_t ilow", "Py_ssize_t ihigh", "PyObject* v"],
+        cmpfunc=unhandled_error_compare
+    )
+
+    def _reference_delslice(args):
+        del args[0][args[1]:args[2]]
+        return 0;
+
+    test_PySequence_DelSlice = CPyExtFunction(
+        _reference_delslice,
+        lambda: (
+            ([1,2,3,4],0,4),
+            ([],1,2),
+            ([1,2,3,4],10,20),
+        ),
+        resultspec="i",
+        argspec='Onn',
+        arguments=["PyObject* op", "Py_ssize_t ilow", "Py_ssize_t ihigh"],
+        cmpfunc=unhandled_error_compare
+    )
+
     test_PySequence_ITEM = CPyExtFunction(
         _reference_getitem,
         lambda: (

graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_dict.py

@@ -60,13 +60,18 @@ def _reference_values(args):
     return list(d.values())
 
 
+def _reference_items(args):
+    d = args[0]
+    return list(d.items())
+
+
 def _reference_pop(args):
     d = args[0]
     if(len(args) == 2):
         return d.pop(args[1])
     else:
         return d.pop(args[1], args[2])
-        
+
 
 def _reference_get_item_with_error(args):
     d = args[0]
@@ -127,7 +132,7 @@ def _reference_merge(args):
                 if not k in a:
                     a[k] = b[k]
         return 0
-    except:        
+    except:
         raise AttributeError("'int' object has no attribute 'keys'")
 
 class SubDict(dict):
@@ -158,7 +163,7 @@ def keys(self):
         return "ab"
     def __getitem__(self, key):
         return key
-        
+
 class TestPyDict(CPyExtTestCase):
 
     def compile_module(self, name):
@@ -174,7 +179,7 @@ def compile_module(self, name):
         arguments=("PyObject* dict", "PyObject* key", "PyObject* deflt"),
         cmpfunc=unhandled_error_compare
     )
-    
+
     # PyDict_SetItem
     test_PyDict_SetItem = CPyExtFunction(
         _reference_set_item,
@@ -208,7 +213,7 @@ def compile_module(self, name):
         callfunction="wrap_PyDict_GetItem",
         cmpfunc=unhandled_error_compare
     )
-    
+
     # PyDict_GetItemWithError
     test_PyDict_GetItemWithError = CPyExtFunction(
         _reference_get_item_with_error,
@@ -344,12 +349,12 @@ def compile_module(self, name):
         lambda: (({'a': "hello"}, ),),
         code='''PyObject* wrap__PyDict_SetItem_KnownHash(PyObject* dict) {
             PyObject* result = PyDict_New();
-            
+
             Py_ssize_t ppos = 0;
             PyObject* key;
             PyObject* value;
-            Py_hash_t phash;            
-            
+            Py_hash_t phash;
+
             _PyDict_Next(dict, &ppos, &key, &value, &phash);
             _PyDict_SetItem_KnownHash(result, key, value, phash);
             return result;
@@ -499,7 +504,7 @@ def compile_module(self, name):
         argspec="O",
         cmpfunc=unhandled_error_compare
     )
-    
+
     # PyDict_Values
     test_PyDict_Values = CPyExtFunction(
         _reference_values,
@@ -508,3 +513,11 @@ def compile_module(self, name):
         argspec="O",
         cmpfunc=unhandled_error_compare
     )
+
+    test_PyDict_Items = CPyExtFunction(
+        _reference_items,
+        lambda: (({},), ({'a': "hello"},)),
+        resultspec="O",
+        argspec="O",
+        cmpfunc=unhandled_error_compare
+    )

graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_exceptionobject.py

@@ -127,6 +127,18 @@ def test_set_exc_info(self):
         else:
             assert False
 
+
+def raise_exception_with_cause():
+    try:
+        raise RuntimeError()
+    except RuntimeError as e:
+        exc = e
+    try:
+        raise IndexError from exc
+    except IndexError as e:
+        return e
+
+
 class TestExceptionobjectFunctions(CPyExtTestCase):
     def compile_module(self, name):
         type(self).mro()[1].__dict__["test_%s" % name].create_module(name)
@@ -143,3 +155,13 @@ def compile_module(self, name):
         argspec="OO",
         arguments=["PyObject* exc", "PyObject* tb"],
     )
+
+    test_PyException_GetCause = CPyExtFunction(
+        lambda args: args[0].__cause__,
+        lambda: (
+            (raise_exception_with_cause(),),
+        ),
+        resultspec="O",
+        argspec="O",
+        arguments=["PyObject* exc"],
+    )

graalpython/com.oracle.graal.python.test/src/tests/cpyext/test_pythonrun.py

@@ -84,3 +84,25 @@ def compile_module(self, name):
         arguments=["char* source", "int type", "PyObject* globals", "PyObject* locals", "PyCompilerFlags* ignored"],
         cmpfunc=unhandled_error_compare
     )
+
+    test_Py_CompileString = CPyExtFunction(
+        lambda args: compile(
+            args[0],
+            args[1],
+            {
+                256: "single",
+                257: "exec",
+                258: "eval"
+            }[args[2]]
+        ),
+        lambda: (
+            ("1 + 2", "foo.py", 256),
+            ("1 + 2", "foo.py", 257),
+            ("1 + 2", "foo.py", 258),
+            ("x = 2", "foo.py", 258),
+        ),
+        resultspec="O",
+        argspec='ssi',
+        arguments=["char* source", "char* filename", "int type"],
+        cmpfunc=unhandled_error_compare
+    )