mirror of
https://github.com/zama-ai/concrete.git
synced 2026-01-13 14:58:01 -05:00
547 lines
9.4 KiB
Python
547 lines
9.4 KiB
Python
"""
|
|
Tests of execution of static assignment operation.
|
|
"""
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from concrete import fhe
|
|
|
|
|
|
def assignment_case_0():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (3,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[:] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_1():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (3,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[0] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_2():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (3,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[1] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_3():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (3,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[2] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_4():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[0:3] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_5():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[1:4] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_6():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[1:4:2] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_7():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (10,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[::2] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_8():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[2:0:-1] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_9():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5,)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[4:0:-2] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_10():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5,)
|
|
value = np.random.randint(0, 2**7, size=(3,))
|
|
|
|
def assign(x):
|
|
x[1:4] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_11():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5,)
|
|
value = np.random.randint(0, 2**7, size=(3,))
|
|
|
|
def assign(x):
|
|
x[4:1:-1] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_12():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (10,)
|
|
value = np.random.randint(0, 2**7, size=(3,))
|
|
|
|
def assign(x):
|
|
x[1:7:2] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_13():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (10,)
|
|
value = np.random.randint(0, 2**7, size=(3,))
|
|
|
|
def assign(x):
|
|
x[7:1:-2] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_14():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[0, 0] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_15():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[3, 1] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_16():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[0] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_17():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(4,))
|
|
|
|
def assign(x):
|
|
x[0] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_18():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(5,))
|
|
|
|
def assign(x):
|
|
x[:, 0] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_19():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(5,))
|
|
|
|
def assign(x):
|
|
x[:, 1] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_20():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[0:3, :] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_21():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(3, 4))
|
|
|
|
def assign(x):
|
|
x[0:3, :] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_22():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(4,))
|
|
|
|
def assign(x):
|
|
x[0:3, :] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_23():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(3,))
|
|
|
|
def assign(x):
|
|
x[0:3, 1:4] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_24():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(3, 3))
|
|
|
|
def assign(x):
|
|
x[0:3, 1:4] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_25():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(3, 3))
|
|
|
|
def assign(x):
|
|
x[4:1:-1, 3:0:-1] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_26():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(3,))
|
|
|
|
def assign(x):
|
|
x[3:0:-1, 0] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_27():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=(2,))
|
|
|
|
def assign(x):
|
|
x[0, 1:3] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_28():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5, 4)
|
|
value = np.random.randint(0, 2**7, size=())
|
|
|
|
def assign(x):
|
|
x[2:4, 1:3] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
def assignment_case_29():
|
|
"""
|
|
Assignment test case.
|
|
"""
|
|
|
|
shape = (5,)
|
|
value = -20
|
|
|
|
def assign(x):
|
|
x[0] = value
|
|
return x
|
|
|
|
return shape, assign
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"shape,function",
|
|
[
|
|
pytest.param(*assignment_case_0()),
|
|
pytest.param(*assignment_case_1()),
|
|
pytest.param(*assignment_case_2()),
|
|
pytest.param(*assignment_case_3()),
|
|
pytest.param(*assignment_case_4()),
|
|
pytest.param(*assignment_case_5()),
|
|
pytest.param(*assignment_case_6()),
|
|
pytest.param(*assignment_case_7()),
|
|
pytest.param(*assignment_case_8()),
|
|
pytest.param(*assignment_case_9()),
|
|
pytest.param(*assignment_case_10()),
|
|
pytest.param(*assignment_case_11()),
|
|
pytest.param(*assignment_case_12()),
|
|
pytest.param(*assignment_case_13()),
|
|
pytest.param(*assignment_case_14()),
|
|
pytest.param(*assignment_case_15()),
|
|
pytest.param(*assignment_case_16()),
|
|
pytest.param(*assignment_case_17()),
|
|
pytest.param(*assignment_case_18()),
|
|
pytest.param(*assignment_case_19()),
|
|
pytest.param(*assignment_case_20()),
|
|
pytest.param(*assignment_case_21()),
|
|
pytest.param(*assignment_case_22()),
|
|
pytest.param(*assignment_case_23()),
|
|
pytest.param(*assignment_case_24()),
|
|
pytest.param(*assignment_case_25()),
|
|
pytest.param(*assignment_case_26()),
|
|
pytest.param(*assignment_case_27()),
|
|
pytest.param(*assignment_case_28()),
|
|
pytest.param(*assignment_case_29()),
|
|
],
|
|
)
|
|
def test_static_assignment(shape, function, helpers):
|
|
"""
|
|
Test static assignment.
|
|
"""
|
|
|
|
configuration = helpers.configuration()
|
|
compiler = fhe.Compiler(function, {"x": "encrypted"})
|
|
|
|
inputset = [np.random.randint(0, 2**7, size=shape) for _ in range(100)]
|
|
circuit = compiler.compile(inputset, configuration)
|
|
|
|
sample = np.random.randint(0, 2**7, size=shape)
|
|
helpers.check_execution(circuit, function, sample)
|
|
|
|
|
|
def test_bad_static_assignment(helpers):
|
|
"""
|
|
Test static assingment with bad parameters.
|
|
"""
|
|
|
|
configuration = helpers.configuration()
|
|
|
|
# with float
|
|
# ----------
|
|
|
|
def f(x):
|
|
x[1.5] = 0
|
|
return x
|
|
|
|
compiler = fhe.Compiler(f, {"x": "encrypted"})
|
|
|
|
inputset = [np.random.randint(0, 2**3, size=(3,)) for _ in range(100)]
|
|
with pytest.raises(ValueError) as excinfo:
|
|
compiler.compile(inputset, configuration)
|
|
|
|
assert str(excinfo.value) == "Assigning to '1.5' is not supported"
|
|
|
|
# with bad slice
|
|
# --------------
|
|
|
|
def g(x):
|
|
x[slice(1.5, 2.5, None)] = 0
|
|
return x
|
|
|
|
compiler = fhe.Compiler(g, {"x": "encrypted"})
|
|
|
|
inputset = [np.random.randint(0, 2**3, size=(3,)) for _ in range(100)]
|
|
with pytest.raises(ValueError) as excinfo:
|
|
compiler.compile(inputset, configuration)
|
|
|
|
assert str(excinfo.value) == "Assigning to '1.5:2.5' is not supported"
|