openfhe-python/tests/test_cryptocontext.py at 83e7fda0f4d654d4e18d3a9cbbc36e3f6da854a9 · openfheorg/openfhe-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import pytest
import openfhe as fhe

@pytest.mark.skipif(fhe.get_native_int() != 128, reason="Only for NATIVE_INT=128")
@pytest.mark.parametrize("scaling", [fhe.FIXEDAUTO, fhe.FIXEDMANUAL])
def test_ckks_context_nativeint128(scaling):
    batch_size = 8
    parameters = fhe.CCParamsCKKSRNS()
    parameters.SetMultiplicativeDepth(5)
    parameters.SetScalingModSize(78)
    parameters.SetBatchSize(batch_size)
    parameters.SetScalingTechnique(scaling)
    parameters.SetNumLargeDigits(2)
    cc = fhe.GenCryptoContext(parameters)
    assert isinstance(cc, fhe.CryptoContext)


@pytest.mark.parametrize("scaling", [fhe.FIXEDAUTO, fhe.FIXEDMANUAL])
def test_ckks_context(scaling):
    batch_size = 8
    parameters = fhe.CCParamsCKKSRNS()
    parameters.SetMultiplicativeDepth(5)
    parameters.SetScalingModSize(60-1)
    parameters.SetBatchSize(batch_size)
    parameters.SetScalingTechnique(scaling)
    parameters.SetNumLargeDigits(2)
    cc = fhe.GenCryptoContext(parameters)
    assert isinstance(cc, fhe.CryptoContext)