github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-08 11:35:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(frontend-python): lazily enable simulation and execution when needed

Browse Source
This commit is contained in:
Umut
2023-06-30 14:29:21 +02:00
parent 6fdcb78158
commit ee5fc138ab
2 changed files with 75 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
frontends/concrete-python/tests/compilation/test_circuit.py
View File

@@ -422,12 +422,8 @@ def test_circuit_sim_disabled(helpers):
inputset = [(np.random.randint(0, 2**4), np.random.randint(0, 2**5)) for _ in range(2)]
circuit = f.compile(inputset, configuration)
with pytest.raises(RuntimeError) as excinfo:
circuit.simulate(*inputset[0])
assert (
str(excinfo.value)
== "Simulation isn't enabled. You can call enable_fhe_simulation() to enable it"
)
assert circuit.simulate(*inputset[0]) == f(*inputset[0])
def test_circuit_fhe_exec_disabled(helpers):
@@ -443,13 +439,8 @@ def test_circuit_fhe_exec_disabled(helpers):
inputset = [(np.random.randint(0, 2**4), np.random.randint(0, 2**5)) for _ in range(2)]
circuit = f.compile(inputset, configuration.fork(fhe_execution=False))
with pytest.raises(RuntimeError) as excinfo:
# as we can't encrypt, we just pass plain inputs, and it should lead to the expected error
circuit.run(*inputset[0], None)
assert (
str(excinfo.value)
== "FHE execution isn't enabled. You can call enable_fhe_execution() to enable it"
)
assert circuit.encrypt_run_decrypt(*inputset[0]) == f(*inputset[0])
def test_circuit_fhe_exec_no_eval_keys(helpers):