diff --git a/test/testcrypt.py b/test/testcrypt.py index 77152429..69706d7a 100644 --- a/test/testcrypt.py +++ b/test/testcrypt.py @@ -190,9 +190,13 @@ def _setup(scope): prefix, suffix = "FUNC(", ")" valprefix = "val_" outprefix = "out_" + optprefix = "opt_" consprefix = "consumed_" def trim_argtype(arg): + if arg.startswith(optprefix): + return optprefix + trim_argtype(arg[len(optprefix):]) + if (arg.startswith(valprefix) and "_" in arg[len(valprefix):]): # Strip suffixes like val_string_asciz