deepcopy stored on module for AK_ArrayDeepCopy

Author: flexatone

src/_arraykit.c

@@ -150,9 +150,9 @@ AK_ResolveDTypeIter(PyObject *dtypes)
     return resolved;
 }
 
-// Perform a deepcopy on an array, using an optional memo dictionary, and specialized to depend on immutable arrays.
+// Perform a deepcopy on an array, using an optional memo dictionary, and specialized to depend on immutable arrays. This depends on the module object to get the deepcopy method.
 PyObject*
-AK_ArrayDeepCopy(PyArrayObject *array, PyObject *memo)
+AK_ArrayDeepCopy(PyObject* m, PyArrayObject *array, PyObject *memo)
 {
     PyObject *id = PyLong_FromVoidPtr((PyObject*)array);
     if (!id) {
@@ -176,12 +176,7 @@ AK_ArrayDeepCopy(PyArrayObject *array, PyObject *memo)
     PyArray_Descr *dtype = PyArray_DESCR(array); // borrowed ref
 
     if (PyDataType_ISOBJECT(dtype)) {
-        PyObject *copy = PyImport_ImportModule("copy");
-        if (!copy) {
-            goto error;
-        }
-        PyObject *deepcopy = PyObject_GetAttrString(copy, "deepcopy");
-        Py_DECREF(copy);
+        PyObject *deepcopy = PyObject_GetAttrString(m, "deepcopy");
         if (!deepcopy) {
             goto error;
         }
@@ -327,7 +322,7 @@ static char *array_deepcopy_kwarg_names[] = {
 
 // Specialized array deepcopy that stores immutable arrays in an optional memo dict that can be provided with kwargs.
 static PyObject *
-array_deepcopy(PyObject *Py_UNUSED(m), PyObject *args, PyObject *kwargs)
+array_deepcopy(PyObject *m, PyObject *args, PyObject *kwargs)
 {
     PyObject *array;
     PyObject *memo = NULL;
@@ -338,7 +333,7 @@ array_deepcopy(PyObject *Py_UNUSED(m), PyObject *args, PyObject *kwargs)
         return NULL;
     }
     AK_CHECK_NUMPY_ARRAY(array);
-    return AK_ArrayDeepCopy((PyArrayObject*)array, memo);
+    return AK_ArrayDeepCopy(m, (PyArrayObject*)array, memo);
 }
 
 //------------------------------------------------------------------------------
@@ -784,11 +779,26 @@ PyInit__arraykit(void)
 {
     import_array();
     PyObject *m = PyModule_Create(&arraykit_module);
+
+    PyObject *copy = PyImport_ImportModule("copy");
+    if (!copy) {
+        Py_DECREF(m);
+        return NULL;
+    }
+    PyObject *deepcopy = PyObject_GetAttrString(copy, "deepcopy");
+    Py_DECREF(copy);
+    if (!deepcopy) {
+        Py_DECREF(m);
+        return NULL;
+    }
+
     if (!m ||
         PyModule_AddStringConstant(m, "__version__", Py_STRINGIFY(AK_VERSION)) ||
         PyType_Ready(&ArrayGOType) ||
-        PyModule_AddObject(m, "ArrayGO", (PyObject *) &ArrayGOType))
+        PyModule_AddObject(m, "ArrayGO", (PyObject *) &ArrayGOType) ||
+        PyModule_AddObject(m, "deepcopy", deepcopy))
     {
+        Py_DECREF(deepcopy);
         Py_XDECREF(m);
         return NULL;
     }

tasks.py

@@ -36,8 +36,8 @@ def test(context):
 
 
 @invoke.task(build)
-def performance(context):
-    context.run(f'{sys.executable} -m performance', echo=True, pty=True)
+def performance(context, names=''):
+    context.run(f'{sys.executable} -m performance {"--names" if names else ""} {names}', echo=True, pty=True)
 
 
 @invoke.task