commit 18f94d8175936781ec2f3d7aa6469f2392404ef3
author Werner Lewis <werner.lewis@arm.com> Wed Aug 31 16:55:44 2022 +0100
committer Werner Lewis <werner.lewis@arm.com> Wed Sep 21 16:49:13 2022 +0100
tree c242fad23866ea36fdba986cb1de746b3550e95f
parent 2b0f7d8d56ae900fcb7fb66f40cc9df37a4af173

Use `combinations_with_replacement` for inputs

When generating combinations of values, `itertools.combinations` will
not allow inputs to be repeated. This is replaced so that cases where
input values match are generated, i.e. ("0", "0").

Signed-off-by: Werner Lewis <werner.lewis@arm.com>

tests/scripts/generate_bignum_tests.py

diff --git a/tests/scripts/generate_bignum_tests.py b/tests/scripts/generate_bignum_tests.py
index 8a8425e..b08ba37 100755
--- a/tests/scripts/generate_bignum_tests.py
+++ b/tests/scripts/generate_bignum_tests.py
@@ -166,7 +166,7 @@
         """
         yield from cast(
             Iterator[Tuple[str, str]],
-            itertools.combinations(cls.input_values, 2)
+            itertools.combinations_with_replacement(cls.input_values, 2)
         )
         yield from cls.input_cases
 
@@ -215,7 +215,7 @@
     test_name = "MPI add"
     input_cases = cast(
         List[Tuple[str, str]],
-        list(itertools.combinations(
+        list(itertools.combinations_with_replacement(
             [
                 "1c67967269c6", "9cde3",
                 "-1c67967269c6", "-9cde3",