mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-11 05:04:58 -05:00
101 lines
3.4 KiB
C++
101 lines
3.4 KiB
C++
#include <gtest/gtest.h>
|
|
|
|
#include "concretelang/ClientLib/ClientParameters.h"
|
|
#include "concretelang/ClientLib/EncryptedArguments.h"
|
|
#include "concretelang/Support/V0Curves.h"
|
|
#include "tests_tools/assert.h"
|
|
|
|
namespace clientlib = concretelang::clientlib;
|
|
|
|
// Define a fixture for instantiate test with client parameters
|
|
class ClientParametersTest
|
|
: public ::testing::TestWithParam<clientlib::ClientParameters> {
|
|
protected:
|
|
clientlib::ClientParameters clientParameters;
|
|
};
|
|
|
|
// Test case encrypt and decrypt
|
|
TEST_P(ClientParametersTest, encrypt_decrypt) {
|
|
|
|
auto clientParameters = GetParam();
|
|
|
|
// Generate the client keySet
|
|
ASSERT_ASSIGN_OUTCOME_VALUE(
|
|
keySet, clientlib::KeySet::generate(clientParameters, 0, 0));
|
|
|
|
// Allocate the ciphertext
|
|
uint64_t *ciphertext = nullptr;
|
|
uint64_t size = 0;
|
|
ASSERT_OUTCOME_HAS_VALUE(keySet->allocate_lwe(0, &ciphertext, size));
|
|
|
|
// Encrypt
|
|
uint64_t input = 3;
|
|
ASSERT_OUTCOME_HAS_VALUE(keySet->encrypt_lwe(0, ciphertext, input));
|
|
|
|
// Decrypt
|
|
uint64_t output;
|
|
ASSERT_OUTCOME_HAS_VALUE(keySet->decrypt_lwe(0, ciphertext, output));
|
|
|
|
ASSERT_EQ(input, output) << "decrypted value differs than the encrypted one";
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
/// Instantiate test suite with generated client parameters ///////////////////
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
/// Create a client parameters with just one secret key of `dimension` and with
|
|
/// one input scalar gate and one output scalar gate on the same key
|
|
clientlib::ClientParameters
|
|
generateClientParameterOneScalarOneScalar(clientlib::LweDimension dimension,
|
|
clientlib::Precision precision) {
|
|
// One secret key with the given dimension
|
|
clientlib::ClientParameters params;
|
|
params.secretKeys.insert({clientlib::SMALL_KEY, {/*.dimension =*/dimension}});
|
|
// One input and output encryption gate on the same secret key and encoded
|
|
// with the same precision
|
|
clientlib::EncryptionGate encryption;
|
|
encryption.secretKeyID = clientlib::SMALL_KEY;
|
|
encryption.encoding.precision = precision;
|
|
clientlib::CircuitGate gate;
|
|
gate.encryption = encryption;
|
|
params.inputs.push_back(gate);
|
|
params.outputs.push_back(gate);
|
|
return params;
|
|
}
|
|
|
|
std::vector<clientlib::ClientParameters> generateAllParameters() {
|
|
// All lwe dimensions to test
|
|
std::vector<clientlib::LweDimension> lweDimensions{
|
|
1 << 9, 1 << 10, 1 << 11, 1 << 12, 1 << 13,
|
|
};
|
|
|
|
// All precision to test
|
|
std::vector<clientlib::Precision> precisions(8, 0);
|
|
llvm::for_each(llvm::enumerate(precisions),
|
|
[](auto p) { p.value() = p.index() + 1; });
|
|
|
|
// All client parameters to test
|
|
std::vector<clientlib::ClientParameters> parameters;
|
|
|
|
for (auto dimension : lweDimensions) {
|
|
for (auto precision : precisions) {
|
|
parameters.push_back(
|
|
generateClientParameterOneScalarOneScalar(dimension, precision));
|
|
}
|
|
}
|
|
|
|
return parameters;
|
|
}
|
|
|
|
INSTANTIATE_TEST_SUITE_P(
|
|
OneScalarOnScalar, ClientParametersTest,
|
|
::testing::ValuesIn(generateAllParameters()),
|
|
[](const testing::TestParamInfo<clientlib::ClientParameters> info) {
|
|
auto cp = info.param;
|
|
auto input_0 = cp.inputs[0];
|
|
return std::string("lweDimension_") +
|
|
std::to_string(cp.lweSecretKeyParam(input_0).value().dimension) +
|
|
"_precision_" +
|
|
std::to_string(input_0.encryption.getValue().encoding.precision);
|
|
});
|