extended test case

diff --git a/tests/run/for_in_string.pyx b/tests/run/for_in_string.pyx
index a892d1a5433e762a10c73d99998fecc04f715807..477b56d06962859c30df02ff3932cb1d1bf247cd 100644 (file)
--- a/tests/run/for_in_string.pyx
+++ b/tests/run/for_in_string.pyx
@@ -2,9 +2,13 @@ cimport cython
 
 bytes_abc = b'abc'
 bytes_ABC = b'ABC'
+bytes_abc_null = b'a\0b\0c'
+bytes_ABC_null = b'A\0B\0C'
 
 unicode_abc = u'abc'
 unicode_ABC = u'ABC'
+unicode_abc_null = u'a\0b\0c'
+unicode_ABC_null = u'A\0B\0C'
 
 
 def for_in_bytes(bytes s):
@@ -13,6 +17,10 @@ def for_in_bytes(bytes s):
     'X'
     >>> for_in_bytes(bytes_ABC)
     'C'
+    >>> for_in_bytes(bytes_abc_null)
+    'X'
+    >>> for_in_bytes(bytes_ABC_null)
+    'C'
     """
     for c in s:
         # Py2/Py3
@@ -29,6 +37,10 @@ def for_char_in_bytes(bytes s):
     'X'
     >>> for_char_in_bytes(bytes_ABC)
     'C'
+    >>> for_char_in_bytes(bytes_abc_null)
+    'X'
+    >>> for_char_in_bytes(bytes_ABC_null)
+    'C'
     """
     cdef char c
     for c in s:
@@ -45,6 +57,10 @@ def for_char_in_enumerate_bytes(bytes s):
     'X'
     >>> for_char_in_enumerate_bytes(bytes_ABC)
     2
+    >>> for_char_in_enumerate_bytes(bytes_abc_null)
+    'X'
+    >>> for_char_in_enumerate_bytes(bytes_ABC_null)
+    4
     """
     cdef char c
     cdef Py_ssize_t i
@@ -62,6 +78,10 @@ def for_pyunicode_in_unicode(unicode s):
     'X'
     >>> for_pyunicode_in_unicode(unicode_ABC)
     'C'
+    >>> for_pyunicode_in_unicode(unicode_abc_null)
+    'X'
+    >>> for_pyunicode_in_unicode(unicode_ABC_null)
+    'C'
     """
     cdef Py_UNICODE c
     for c in s:
@@ -78,6 +98,10 @@ def for_pyunicode_in_enumerate_unicode(unicode s):
     'X'
     >>> for_pyunicode_in_enumerate_unicode(unicode_ABC)
     2
+    >>> for_pyunicode_in_enumerate_unicode(unicode_abc_null)
+    'X'
+    >>> for_pyunicode_in_enumerate_unicode(unicode_ABC_null)
+    4
     """
     cdef Py_UNICODE c
     cdef Py_ssize_t i