Merge pull request #195 from djkirkham/bugfix-arraystack-deepcopy

Replace vectorize with list comprehension in ArrayStack.__deepcopy__

biggus/_init.py

@@ -1757,10 +1757,10 @@ def __deepcopy__(self, memo):
 
         # Preserve the array ordering by performing an
         # element-wise deepcopy of the stack payload.
-        deepcopy_with_memo = functools.partial(deepcopy, memo=memo)
-        deepcopy_elementwise = np.vectorize(deepcopy_with_memo,
-                                            otypes=[np.object])
-        result = deepcopy_elementwise(self._stack)
+        result = np.array(
+            [deepcopy(array, memo=memo) for array in self._stack.flat],
+            dtype='O')
+        result.shape = self._stack.shape
         memo[id(self._stack)] = result
 
         # Implement a standard deepcopy now we've already dealt

biggus/tests/unit/init/test_ArrayStack.py

@@ -199,6 +199,8 @@ def check(self, order):
                                    order=order, dtype=object))
         copied = copy.deepcopy(orig)
         assert_array_equal(expected, copied.ndarray())
+        for array in copied._stack.flat:
+            self.assertTrue(isinstance(array, NumpyArrayAdapter))
 
     def test_fortran_order(self):
         self.check('f')