Generate: warn for count>100; error and limit for count>1000

RagnarGrootKoerkamp · Nov 16, 2024 · 373db10 · 373db10
1 parent a486075
commit 373db10
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/bin/generate.py b/bin/generate.py
@@ -1508,15 +1508,25 @@ def parse_count(yaml, warn_for=None):
                         color_type=MessageType.WARN,
                     )
                 return 1
+            if count > 1000:
+                if warn_for is not None:
+                    message(
+                        f'Found count: {count}, limited to 1000.',
+                        'generators.yaml',
+                        warn_for,
+                        color_type=MessageType.ERROR,
+                    )
+                return 1000
             if count > 100:
                 if warn_for is not None:
                     message(
-                        f'Found count: {count}, limited to 100.',
+                        f'Found large count: {count}.',
                         'generators.yaml',
                         warn_for,
                         color_type=MessageType.WARN,
                     )
-                return 100
+                return count
+
             return count
 
         # Count the number of testcases in the given directory yaml.