From 98fc1f179b6069ecd05eb99ec71d2ec53cf2becb Mon Sep 17 00:00:00 2001 From: Gilles Peskine Date: Wed, 8 Jul 2026 12:51:27 +0200 Subject: [PATCH] Support multiple private key formats in DriverGenerator Signed-off-by: Gilles Peskine --- util/mbedtls_maintainer/mldsa_test_generator.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/util/mbedtls_maintainer/mldsa_test_generator.py b/util/mbedtls_maintainer/mldsa_test_generator.py index 615c41b57..70e09ab5e 100644 --- a/util/mbedtls_maintainer/mldsa_test_generator.py +++ b/util/mbedtls_maintainer/mldsa_test_generator.py @@ -394,8 +394,13 @@ class DriverGenerator(Generator): yield self.one_multipart('verify', key, lengths, tweak_signature=tweak) - def gen_all(self, multipart: bool = False) -> Iterator[test_case.TestCase]: + def gen_all(self, + multipart: bool = False, + private_key_formats: Optional[Sequence[PrivateKeyFormat]] = None, + ) -> Iterator[test_case.TestCase]: """Generate all the tests for this API.""" + if private_key_formats is not None: + self.private_key_formats = private_key_formats yield from super().gen_all() if multipart: for kl in sorted(KEYS.keys()):