Mod operations: fill arguments to the width of the modulus

With the default input style (which is "variable"), fill all bignum test
case arguments to the same width as the modulus.

Signed-off-by: Gilles Peskine <Gilles.Peskine@arm.com>
This commit is contained in:
Gilles Peskine 2022-12-20 19:16:54 +01:00
parent f8a4463bd6
commit 5623ecc2d6

View file

@ -267,6 +267,12 @@ class ModOperationCommon(OperationCommon):
def arg_n(self) -> str:
return self.format_arg(self.val_n)
def format_arg(self, val: str) -> str:
if self.input_style == "variable":
return val.zfill(len(hex(self.int_n)) - 2)
else:
return super().format_arg(val)
def arguments(self) -> List[str]:
return [quote_str(self.arg_n)] + super().arguments()