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

test: fix ldexp correctness test by changing the input type

Browse Source
This commit is contained in:
Arthur Meyre
2021-10-15 10:01:25 +02:00
parent 73769b917e
commit a811b588c6
1 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/numpy/test_compile.py
View File

@@ -164,7 +164,7 @@ def subtest_compile_and_run_binary_ufunc_correctness(ufunc, upper_function, c, i
for prod in itertools.product(*args):
yield prod
function_parameters = {arg_name: EncryptedScalar(Integer(64, False)) for arg_name in ["x", "y"]}
function_parameters = {arg_name: EncryptedScalar(Integer(64, True)) for arg_name in ["x", "y"]}
compiler_engine = compile_numpy_function(
function,
@@ -222,14 +222,10 @@ def test_binary_ufunc_operations(ufunc):
ufunc, mix_x_and_y_and_call_binary_f_two, 2, ((0, 5), (0, 5))
)
elif ufunc in [numpy.ldexp]:
# Can't make it work
# TODO: fixme
pass
# Need small constants to keep results sufficiently small
# subtest_compile_and_run_binary_ufunc_correctness(
# ufunc, mix_x_and_y_and_call_binary_f_two, 2, ((0, 5), (0, 5))
# )
subtest_compile_and_run_binary_ufunc_correctness(
ufunc, mix_x_and_y_and_call_binary_f_two, 2, ((0, 5), (0, 5))
)
else:
# General case
subtest_compile_and_run_binary_ufunc_correctness(