commit 77b877d5a79277e445a15aed73b5c735fa074c63
author Gabor Mezei <gabor.mezei@arm.com> Fri Dec 16 15:25:02 2022 +0100
committer Gabor Mezei <gabor.mezei@arm.com> Tue Dec 20 17:30:19 2022 +0100
tree dbdaea4705075a108c9f33bade534ee7c7c82604
parent 2840884c35e966fe7ebc97576feccaecf6509cee

Generate operands in Mongomery representation for the test function

Signed-off-by: Gabor Mezei <gabor.mezei@arm.com>

scripts/mbedtls_dev/bignum_mod.py

diff --git a/scripts/mbedtls_dev/bignum_mod.py b/scripts/mbedtls_dev/bignum_mod.py
index e0fc159..1960723 100644
--- a/scripts/mbedtls_dev/bignum_mod.py
+++ b/scripts/mbedtls_dev/bignum_mod.py
@@ -41,14 +41,14 @@
     arity = 2
 
     def arguments(self) -> List[str]:
-        return [bignum_common.quote_str(n) for n in [self.arg_a,
-                                                     self.arg_b,
-                                                     self.arg_n]
+        return [self.format_result(self.to_montgomery(self.int_a)),
+                self.format_result(self.to_montgomery(self.int_b)),
+                bignum_common.quote_str(self.arg_n)
                ] + self.result()
 
     def result(self) -> List[str]:
         result = (self.int_a * self.int_b) % self.int_n
-        return [self.format_result(result)]
+        return [self.format_result(self.to_montgomery(result))]
 
 # END MERGE SLOT 2