Fixed optional bug in invert

smartycope · Apr 20, 2024 · 6e138e7 · 6e138e7
1 parent b6f5801
commit 6e138e7
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/ezregex/invert.py b/ezregex/invert.py
@@ -160,7 +160,11 @@ def handle(pattern, amt=1, opposite=False):
                         case sre.RANGE:
                             start, end = args
                             s += chr(randint(start, end))
-                        case sre.MAX_REPEAT | sre.MIN_REPEAT:
+                        case sre.MIN_REPEAT: # optional, I believe
+                            min, max, sub = args
+                            if randint(0, 1) == 1:
+                                s += handle(sub, amt)
+                        case sre.MAX_REPEAT:
                             min, max, sub = args
                             if max is None or max is sre.MAXREPEAT:
                                 max = randint(min, self.alot)

diff --git a/tests/data/regexs.jsonc b/tests/data/regexs.jsonc
@@ -966,6 +966,11 @@
         [],
         []
     ],
+    [
+        "group('pattern', name='i') + optional(digit) + 'test'",
+        [],
+        []
+    ],
     // TODO: This is for bytes only programs or something?...
     // [
     //     "digit + LOCALE",