]> granicus.if.org Git - python/commitdiff
Array module's buffer interface can now handle empty arrays.
authorRaymond Hettinger <python@rcn.com>
Mon, 2 Apr 2007 22:54:21 +0000 (22:54 +0000)
committerRaymond Hettinger <python@rcn.com>
Mon, 2 Apr 2007 22:54:21 +0000 (22:54 +0000)
Lib/test/test_re.py
Modules/arraymodule.c

index bb568612fa795e0b40f7c306d6da12fe7c162545..b187ea10dabae9b7785d5f5bdd152015077b9383 100644 (file)
@@ -601,6 +601,13 @@ class ReTests(unittest.TestCase):
         self.assertEqual(iter.next().span(), (4, 4))
         self.assertRaises(StopIteration, iter.next)
 
+    def test_empty_array(self):
+        # SF buf 1647541
+        import array
+        for typecode in 'cbBuhHiIlLfd':
+            a = array.array(typecode)
+            self.assertEqual(re.compile("bla").match(a), None)
+            self.assertEqual(re.compile("").match(a).groups(), ())            
 
 def run_re_tests():
     from test.re_tests import benchmarks, tests, SUCCEED, FAIL, SYNTAX_ERROR
index 0aeb64eeeee6d3056129e05e4ab66746432a8fa1..3ba5cf88e16c8d9f244664d9e4191221f608fb49 100644 (file)
@@ -1745,6 +1745,8 @@ static PyMappingMethods array_as_mapping = {
        (objobjargproc)array_ass_subscr
 };
 
+static const void *emptybuf = "";
+
 static Py_ssize_t
 array_buffer_getreadbuf(arrayobject *self, Py_ssize_t index, const void **ptr)
 {
@@ -1754,6 +1756,8 @@ array_buffer_getreadbuf(arrayobject *self, Py_ssize_t index, const void **ptr)
                return -1;
        }
        *ptr = (void *)self->ob_item;
+       if (*ptr == NULL)
+               *ptr = emptybuf;
        return self->ob_size*self->ob_descr->itemsize;
 }
 
@@ -1766,6 +1770,8 @@ array_buffer_getwritebuf(arrayobject *self, Py_ssize_t index, const void **ptr)
                return -1;
        }
        *ptr = (void *)self->ob_item;
+       if (*ptr == NULL)
+               *ptr = emptybuf;
        return self->ob_size*self->ob_descr->itemsize;
 }