Просмотр исходного кода

cryptocontext check on smaller native ints as well

Muthu Annamalai 1 год назад
Родитель
Сommit
db494ddaaf
1 измененных файлов с 14 добавлено и 2 удалено
  1. 14 2
      tests/test_cryptocontext.py

+ 14 - 2
tests/test_cryptocontext.py

@@ -1,14 +1,26 @@
 import pytest
 import openfhe as fhe
 
-pytestmark = pytest.mark.skipif(fhe.get_native_int() != 128, reason="Only for NATIVE_INT=128")
+@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(78)
+    parameters.SetScalingModSize(60-1)
     parameters.SetBatchSize(batch_size)
     parameters.SetScalingTechnique(scaling)
     parameters.SetNumLargeDigits(2)