additional tests of array2d_tuple_iter

Author: flexatone

src/_arraykit.c

@@ -3633,7 +3633,7 @@ A2DTuple_iternext(A2DTupleObject *self) {
             return NULL;
         }
         for (npy_intp j = 0; j < num_cols; ++j) {
-            // cannot assume input_array is contiguous
+            // cannot assume array is contiguous
             item = PyArray_ToScalar(PyArray_GETPTR2(array, i, j), array);
             if (item == NULL) {
                 Py_DECREF(tuple);

test/test_util.py

@@ -350,6 +350,28 @@ def test_array2d_tuple_iter_d(self) -> None:
         with self.assertRaises(TypeError):
             reversed(it)
 
+    def test_array2d_tuple_iter_e(self) -> None:
+        a1 = np.array([[None, 'bb'], [None, 'dd'], [3, None]])
+        it = array2d_tuple_iter(a1)
+        del a1
+        self.assertEqual(list(it), [(None, 'bb'), (None, 'dd'), (3, None)])
+
+    def test_array2d_tuple_iter_f(self) -> None:
+        a1 = np.array([[None, 'bb'], [None, 'dd'], [3, None]])
+        it1 = array2d_tuple_iter(a1)
+        del a1
+        it2 = iter(it1)
+        self.assertEqual(list(it1), [(None, 'bb'), (None, 'dd'), (3, None)])
+        self.assertEqual(list(it2), []) # expected behavior
+
+    def test_array2d_tuple_iter_g(self) -> None:
+        a1 = np.array([[None, 'bb'], [None, 'dd'], [3, None]])
+        it1 = array2d_tuple_iter(a1)
+        it2 = array2d_tuple_iter(a1)
+        del a1
+        self.assertEqual(list(it1), [(None, 'bb'), (None, 'dd'), (3, None)])
+        self.assertEqual(list(it2), [(None, 'bb'), (None, 'dd'), (3, None)])
+
     #---------------------------------------------------------------------------
 
     def test_isna_element_a(self) -> None: