some pattern builder tests

r3w0p · Apr 24, 2022 · 74d1b9c · 74d1b9c
1 parent c04c48d
commit 74d1b9c
Show file tree

Hide file tree

Showing 2 changed files with 393 additions and 94 deletions.
diff --git a/bobocep/pattern/bobo_pattern_builder.py b/bobocep/pattern/bobo_pattern_builder.py
@@ -61,13 +61,15 @@ def followed_by(self,
                     group: str,
                     predicate: BoboPredicate,
                     times: int = 1,
+                    loop: bool = False,
                     optional: bool = False) -> 'BoboPatternBuilder':
 
         for _ in range(max(times, 1)):
             self._blocks.append(BoboPatternBlock(
                 group=group,
                 predicates=[predicate],
                 strict=False,
+                loop=loop,
                 negated=False,
                 optional=optional))
         return self
@@ -90,13 +92,15 @@ def followed_by_any(self,
                         group: str,
                         predicates: List[BoboPredicate],
                         times: int = 1,
+                        loop: bool = False,
                         optional: bool = False) -> 'BoboPatternBuilder':
 
         for _ in range(max(times, 1)):
             self._blocks.append(BoboPatternBlock(
                 group=group,
                 predicates=predicates,
                 strict=False,
+                loop=loop,
                 negated=False,
                 optional=optional))
         return self