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

chore: fix flaky tests

Browse Source 
closes #997
This commit is contained in:
Benoit Chevallier-Mames
2021-11-23 17:38:15 +01:00
committed by Benoit Chevallier
parent ec396effb2
commit 33064efc06
1 changed files with 36 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
tests/numpy/test_compile.py
View File

@@ -1828,16 +1828,6 @@ def test_wrong_inputs(default_compilation_configuration):
[
pytest.param(lambda x: (x + (-27)) + 32, ((0, 10),), ["x"]),
pytest.param(lambda x: ((-3) * x) + (100 - (x + 1)), ((0, 10),), ["x"]),
pytest.param(
lambda x: (20 + 10 * numpy.tanh(50 * (numpy.cos(x + 33.0)))).astype(numpy.uint32),
((0, 31),),
["x"],
),
pytest.param(
lambda x: (20 * (numpy.cos(x + 33.0)) + 30).astype(numpy.uint32),
((0, 31),),
["x"],
),
pytest.param(
lambda x, y: (-1) * x + (-2) * y + 40,
(
@@ -1869,6 +1859,42 @@ def test_compile_and_run_correctness_with_negative_values(
assert compiler_engine.run(*args) == function(*args)
@pytest.mark.parametrize(
"function,input_ranges,list_of_arg_names",
[
pytest.param(
lambda x: (20 + 10 * numpy.tanh(50 * (numpy.cos(x + 33.0)))).astype(numpy.uint32),
((0, 31),),
["x"],
),
pytest.param(
lambda x: (20 * (numpy.cos(x + 33.0)) + 30).astype(numpy.uint32),
((0, 31),),
["x"],
),
],
)
def test_compile_and_run_correctness_with_negative_values_and_pbs(
function, input_ranges, list_of_arg_names, default_compilation_configuration
):
"""Test correctness of results when running a compiled function, which has some negative
intermediate values."""
function_parameters = {
arg_name: EncryptedScalar(Integer(64, False)) for arg_name in list_of_arg_names
}
compiler_engine = compile_numpy_function(
function,
function_parameters,
data_gen(tuple(range(x[0], x[1] + 1) for x in input_ranges)),
default_compilation_configuration,
)
args = [random.randint(low, high) for (low, high) in input_ranges]
check_is_good_execution(compiler_engine, function, args, verbose=False)
def check_equality_modulo(a, b, modulus):
"""Check that (a mod modulus) == (b mod modulus)"""
return (a % modulus) == (b % modulus)