Skip to content

Commit db494dd

Browse files
ai-mannamalaidsuponitskiy-duality
authored andcommitted
cryptocontext check on smaller native ints as well
1 parent 16aa39d commit db494dd

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

tests/test_cryptocontext.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,26 @@
11
import pytest
22
import openfhe as fhe
33

4-
pytestmark = pytest.mark.skipif(fhe.get_native_int() != 128, reason="Only for NATIVE_INT=128")
4+
@pytest.mark.skipif(fhe.get_native_int() != 128, reason="Only for NATIVE_INT=128")
5+
@pytest.mark.parametrize("scaling", [fhe.FIXEDAUTO, fhe.FIXEDMANUAL])
6+
def test_ckks_context_nativeint128(scaling):
7+
batch_size = 8
8+
parameters = fhe.CCParamsCKKSRNS()
9+
parameters.SetMultiplicativeDepth(5)
10+
parameters.SetScalingModSize(78)
11+
parameters.SetBatchSize(batch_size)
12+
parameters.SetScalingTechnique(scaling)
13+
parameters.SetNumLargeDigits(2)
14+
cc = fhe.GenCryptoContext(parameters)
15+
assert isinstance(cc, fhe.CryptoContext)
16+
517

618
@pytest.mark.parametrize("scaling", [fhe.FIXEDAUTO, fhe.FIXEDMANUAL])
719
def test_ckks_context(scaling):
820
batch_size = 8
921
parameters = fhe.CCParamsCKKSRNS()
1022
parameters.SetMultiplicativeDepth(5)
11-
parameters.SetScalingModSize(78)
23+
parameters.SetScalingModSize(60-1)
1224
parameters.SetBatchSize(batch_size)
1325
parameters.SetScalingTechnique(scaling)
1426
parameters.SetNumLargeDigits(2)

0 commit comments

Comments
 (0)