From a811b588c6a249880fcbafc6c1869bfb0334424a Mon Sep 17 00:00:00 2001 From: Arthur Meyre Date: Fri, 15 Oct 2021 10:01:25 +0200 Subject: [PATCH] test: fix ldexp correctness test by changing the input type --- tests/numpy/test_compile.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/tests/numpy/test_compile.py b/tests/numpy/test_compile.py index 5a4662b4d..b2f1aa847 100644 --- a/tests/numpy/test_compile.py +++ b/tests/numpy/test_compile.py @@ -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(