test_cryptocontext.py 582 B

1234567891011121314151617
  1. import pytest
  2. import openfhe as fhe
  3. @pytest.mark.long
  4. @pytest.mark.skipif(fhe.get_native_int() < 80, reason="Only for NATIVE_INT=128")
  5. @pytest.mark.parametrize("scaling", [fhe.FIXEDAUTO, fhe.FIXEDMANUAL])
  6. def test_ckks_context(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)