Skip to content

Commit 46d1fe3

Browse files
committed
Add feature to set random seed
1 parent 0bc8c43 commit 46d1fe3

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

src/zkregex_fuzzer/cli.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,11 @@ def fuzz_parser():
4747
choices=list(VALID_INPUT_GENERATORS.keys()),
4848
help=f"The valid input generator to use for the fuzzer (options: {list(VALID_INPUT_GENERATORS.keys())})."
4949
)
50+
parser.add_argument(
51+
"--seed",
52+
default=str(uuid.uuid4()),
53+
help=f"Seed for random generator (default: UUIDv4)"
54+
)
5055
parser.add_argument(
5156
"--save",
5257
choices=[status.name for status in HarnessStatus],
@@ -206,9 +211,7 @@ def do_fuzz(args):
206211
kwargs = vars(args)
207212

208213
# set global seed
209-
seed = str(uuid.uuid4())
210-
kwargs['seed'] = seed
211-
random.seed(seed)
214+
random.seed(args.seed)
212215

213216
if args.fuzzer == "grammar":
214217
fuzz_with_grammar(

0 commit comments

Comments
 (0)