diff --git a/tfhe/src/shortint/server_key/tests/noise_distribution/br_dp_packingks_ms.rs b/tfhe/src/shortint/server_key/tests/noise_distribution/br_dp_packingks_ms.rs index 158f55dd0..12affdc24 100644 --- a/tfhe/src/shortint/server_key/tests/noise_distribution/br_dp_packingks_ms.rs +++ b/tfhe/src/shortint/server_key/tests/noise_distribution/br_dp_packingks_ms.rs @@ -13,14 +13,14 @@ use crate::shortint::client_key::ClientKey; use crate::shortint::engine::ShortintEngine; use crate::shortint::list_compression::{CompressionKey, CompressionPrivateKeys}; use crate::shortint::parameters::test_params::{ - TEST_COMP_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - TEST_PARAM_MESSAGE_2_CARRY_2_KS32_PBS_TUNIFORM_2M128, - TEST_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, + TEST_META_PARAM_CPU_2_2_KS32_PBS_PKE_TO_SMALL_ZKV2_TUNIFORM_2M128, + TEST_META_PARAM_CPU_2_2_KS_PBS_PKE_TO_SMALL_ZKV2_TUNIFORM_2M128, }; use crate::shortint::parameters::{ AtomicPatternParameters, CarryModulus, CiphertextModulusLog, CompressionParameters, - MessageModulus, Variance, + MessageModulus, MetaParameters, Variance, }; +use crate::shortint::server_key::tests::parameterized_test::create_parameterized_test; use crate::shortint::server_key::ServerKey; use crate::shortint::{PaddingBit, ShortintEncoding}; use rayon::prelude::*; @@ -96,11 +96,11 @@ where (res, packing_result, ms_result) } -fn sanity_check_encrypt_br_dp_packing_ks_ms

(params: P, comp_params: CompressionParameters) -where - P: Into, -{ - let params: AtomicPatternParameters = params.into(); +fn sanity_check_encrypt_br_dp_packing_ks_ms(meta_params: MetaParameters) { + let (params, comp_params) = ( + meta_params.compute_parameters, + meta_params.compression_parameters.unwrap(), + ); let cks = ClientKey::new(params); let sks = ServerKey::new(&cks); let compression_private_key = cks.new_compression_private_key(comp_params); @@ -158,23 +158,10 @@ where assert_eq!(after_ms.as_view(), extracted.as_view()); } -#[test] -fn test_sanity_check_encrypt_br_dp_packing_ks_ms_test_param_message_2_carry_2_ks_pbs_tuniform_2m128( -) { - sanity_check_encrypt_br_dp_packing_ks_ms( - TEST_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - TEST_COMP_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - ) -} - -#[test] -fn test_sanity_check_encrypt_br_dp_packing_ks_ms_test_param_message_2_carry_2_ks32_pbs_tuniform_2m128( -) { - sanity_check_encrypt_br_dp_packing_ks_ms( - TEST_PARAM_MESSAGE_2_CARRY_2_KS32_PBS_TUNIFORM_2M128, - TEST_COMP_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - ) -} +create_parameterized_test!(sanity_check_encrypt_br_dp_packing_ks_ms { + TEST_META_PARAM_CPU_2_2_KS_PBS_PKE_TO_SMALL_ZKV2_TUNIFORM_2M128, + TEST_META_PARAM_CPU_2_2_KS32_PBS_PKE_TO_SMALL_ZKV2_TUNIFORM_2M128, +}); #[allow(clippy::type_complexity)] fn encrypt_br_dp_packing_ks_ms_inner_helper( @@ -403,11 +390,11 @@ fn encrypt_br_dp_packing_ks_ms_pfail_helper( after_ms } -fn noise_check_encrypt_br_dp_packing_ks_ms_noise

(params: P, comp_params: CompressionParameters) -where - P: Into, -{ - let params: AtomicPatternParameters = params.into(); +fn noise_check_encrypt_br_dp_packing_ks_ms_noise(meta_params: MetaParameters) { + let (params, comp_params) = ( + meta_params.compute_parameters, + meta_params.compression_parameters.unwrap(), + ); let cks = ClientKey::new(params); let sks = ServerKey::new(&cks); let compression_private_key = cks.new_compression_private_key(comp_params); @@ -530,31 +517,17 @@ where assert!(before_ms_normality.null_hypothesis_is_valid && after_ms_is_ok); } +create_parameterized_test!(noise_check_encrypt_br_dp_packing_ks_ms_noise { + TEST_META_PARAM_CPU_2_2_KS_PBS_PKE_TO_SMALL_ZKV2_TUNIFORM_2M128, + TEST_META_PARAM_CPU_2_2_KS32_PBS_PKE_TO_SMALL_ZKV2_TUNIFORM_2M128, +}); -#[test] -fn test_noise_check_encrypt_br_dp_packing_ks_ms_noise_test_param_message_2_carry_2_ks_pbs_tuniform_2m128( -) { - noise_check_encrypt_br_dp_packing_ks_ms_noise( - TEST_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - TEST_COMP_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - ) -} - -#[test] -fn test_noise_check_encrypt_br_dp_packing_ks_ms_noise_test_param_message_2_carry_2_ks32_pbs_tuniform_2m128( -) { - noise_check_encrypt_br_dp_packing_ks_ms_noise( - TEST_PARAM_MESSAGE_2_CARRY_2_KS32_PBS_TUNIFORM_2M128, - TEST_COMP_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - ) -} - -fn noise_check_encrypt_br_dp_packing_ks_ms_pfail

(params: P, comp_params: CompressionParameters) -where - P: Into, -{ - let (pfail_test_meta, params) = { - let mut params: AtomicPatternParameters = params.into(); +fn noise_check_encrypt_br_dp_packing_ks_ms_pfail(meta_params: MetaParameters) { + let (pfail_test_meta, params, comp_params) = { + let (mut params, comp_params) = ( + meta_params.compute_parameters, + meta_params.compression_parameters.unwrap(), + ); let original_message_modulus = params.message_modulus(); let original_carry_modulus = params.carry_modulus(); @@ -666,7 +639,7 @@ where ) }; - (pfail_test_meta, params) + (pfail_test_meta, params, comp_params) }; let cks = ClientKey::new(params); @@ -709,20 +682,7 @@ where pfail_check(&pfail_test_meta, test_result); } -#[test] -fn test_noise_check_encrypt_br_dp_packing_ks_ms_pfail_test_param_message_2_carry_2_ks_pbs_tuniform_2m128( -) { - noise_check_encrypt_br_dp_packing_ks_ms_pfail( - TEST_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - TEST_COMP_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - ) -} - -#[test] -fn test_noise_check_encrypt_br_dp_packing_ks_ms_pfail_test_param_message_2_carry_2_ks32_pbs_tuniform_2m128( -) { - noise_check_encrypt_br_dp_packing_ks_ms_pfail( - TEST_PARAM_MESSAGE_2_CARRY_2_KS32_PBS_TUNIFORM_2M128, - TEST_COMP_PARAM_MESSAGE_2_CARRY_2_KS_PBS_TUNIFORM_2M128, - ) -} +create_parameterized_test!(noise_check_encrypt_br_dp_packing_ks_ms_pfail { + TEST_META_PARAM_CPU_2_2_KS_PBS_PKE_TO_SMALL_ZKV2_TUNIFORM_2M128, + TEST_META_PARAM_CPU_2_2_KS32_PBS_PKE_TO_SMALL_ZKV2_TUNIFORM_2M128, +});