From 825d608f9b7d745acdf3bbe9d79aa5947939a014 Mon Sep 17 00:00:00 2001 From: Tony Arcieri Date: Fri, 30 Jan 2026 12:20:27 -0700 Subject: [PATCH] ml-kem: fix Wycheproof After merging #207 we need to handle errors from invalid hashes --- ml-kem/tests/wycheproof.rs | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/ml-kem/tests/wycheproof.rs b/ml-kem/tests/wycheproof.rs index 1fe95bd..d390f53 100644 --- a/ml-kem/tests/wycheproof.rs +++ b/ml-kem/tests/wycheproof.rs @@ -235,7 +235,12 @@ macro_rules! mlkem_decaps_test { }; #[allow(deprecated)] - let dk = DecapsulationKey::from_expanded(&test_dk).expect("should be valid"); + let dk_result = DecapsulationKey::from_expanded(&test_dk); + if dk_result.is_err() { + assert_eq!(test.result, ExpectedResult::Invalid); + continue; + } + let dk = dk_result.unwrap(); let test_c: EncodedCiphertext = match decode_optional_hex(&test.c, "c") { Some(dk) => dk,