[Bug #1177831] Exercise (?(id)yes|no) for a group other than the first one

diff --git a/Lib/test/test_re.py b/Lib/test/test_re.py
index 07bc63b277a76568ef34f303f549ec4d0d12aa9d..c86f502c50d483ded3a456abbb57e46b6325506b 100644 (file)
--- a/Lib/test/test_re.py
+++ b/Lib/test/test_re.py
@@ -235,6 +235,16 @@ class ReTests(unittest.TestCase):
         self.assertEqual(re.match('^(?:(a)|c)((?(1)|d))$', 'a').groups(),
                          ('a', ''))
 
+       # Tests for bug #1177831: exercise groups other than the first group
+       p = re.compile('(?P<g1>a)(?P<g2>b)?((?(g2)c|d))')
+       self.assertEqual(p.match('abc').groups(),
+                        ('a', 'b', 'c'))
+       self.assertEqual(p.match('ad').groups(),
+                        ('a', None, 'd'))
+       self.assertEqual(p.match('abd'), None)
+       self.assertEqual(p.match('ac'), None)
+       
+
     def test_re_groupref(self):
         self.assertEqual(re.match(r'^(\|)?([^()]+)\1$', '|a|').groups(),
                          ('|', 'a'))