Add tests for newly added C functions

Author: msimacek

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

@@ -68,6 +68,11 @@ def _reference_setitem(args):
     return listObj
 
 
+def _reference_reverse(args):
+    args[0].reverse()
+    return args[0]
+
+
 def _reference_SET_ITEM(args):
     listObj = args[0]
     pos = args[1]
@@ -323,3 +328,25 @@ def compile_module(self, name):
         arguments=["PyObject* o"],
         cmpfunc=unhandled_error_compare
     )
+
+    test_PyList_Reverse = CPyExtFunction(
+        _reference_reverse,
+        lambda: (
+            ([],),
+            ([1, 2, 3],),
+            ([1, "a", 0.1],),
+        ),
+        code='''PyObject* wrap_PyList_Reverse(PyObject* list) {
+            if (PyList_Reverse(list)) {
+                return NULL;
+            }
+            Py_INCREF(list);
+            return list;
+        }
+        ''',
+        resultspec="O",
+        argspec='O',
+        arguments=["PyObject* list"],
+        callfunction="wrap_PyList_Reverse",
+        cmpfunc=unhandled_error_compare
+    )

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

@@ -57,6 +57,12 @@ def _reference_contains(args):
     except TypeError:
         return raise_Py6_SystemError()
 
+
+def _reference_clear(args):
+    args[0].clear()
+    return args[0]
+
+
 class FrozenSetSubclass(frozenset):
     pass
 
@@ -202,3 +208,24 @@ def compile_module(self, name):
         cmpfunc=unhandled_error_compare
     )
 
+    test_PySet_Clear = CPyExtFunction(
+        _reference_clear,
+        lambda: (
+            ({1, 2, 3},),
+            ({1, "a", 0.1},),
+        ),
+        code='''PyObject* wrap_PySet_Clear(PyObject* set) {
+            if (PySet_Clear(set)) {
+                return NULL;
+            }
+            Py_INCREF(set);
+            return set;
+        }
+        ''',
+        resultspec="O",
+        argspec='O',
+        arguments=["PyObject* set"],
+        callfunction="wrap_PySet_Clear",
+        cmpfunc=unhandled_error_compare
+    )
+

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

@@ -93,6 +93,12 @@ def _reference_readchar(args):
     return ord(s[i])
 
 
+def _reference_contains(args):
+    if not isinstance(args[0], str) or not isinstance(args[1], str):
+        raise TypeError
+    return args[1] in args[0]
+
+
 class CustomString(str):
     pass
 
@@ -635,4 +641,16 @@ def compile_module(self, name):
         cmpfunc=unhandled_error_compare
     )
 
+    test_PyUnicode_Contains = CPyExtFunction(
+        _reference_contains,
+        lambda: (
+            ("aaa", "bbb"),
+            ("aaa", "a"),
+        ),
+        resultspec="i",
+        argspec='OO',
+        arguments=["PyObject* haystack", "PyObject* needle"],
+        cmpfunc=unhandled_error_compare
+    )
+