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

fix(python-bindings): Support np.array with dtype upt to 64 bits

Browse Source
This commit is contained in:
Quentin Bourgerie
2022-08-19 11:58:52 +02:00
parent d647bc735f
commit 9257404f5f
5 changed files with 154 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
compiler/tests/python/test_argument_support.py
View File

@@ -43,30 +43,30 @@ def test_accepted_ints(value):
# TODO: #495
# @pytest.mark.parametrize(
# "dtype",
# [
# pytest.param(np.uint8, id="uint8"),
# pytest.param(np.uint16, id="uint16"),
# pytest.param(np.uint32, id="uint32"),
# pytest.param(np.uint64, id="uint64"),
# ],
# )
# def test_accepted_ndarray(dtype):
# value = np.array([0, 1, 2], dtype=dtype)
# try:
# arg = ClientSupport._create_lambda_argument(value)
# except Exception:
# pytest.fail(f"value of type {type(value)} should be supported")
@pytest.mark.parametrize(
"dtype, maxvalue",
[
pytest.param(np.uint8, 2**8 - 1, id="uint8"),
pytest.param(np.uint16, 2**16 - 1, id="uint16"),
pytest.param(np.uint32, 2**32 - 1, id="uint32"),
pytest.param(np.uint64, 2**64 - 1, id="uint64"),
],
)
def test_accepted_ndarray(dtype, maxvalue):
value = np.array([0, 1, 2, maxvalue], dtype=dtype)
try:
arg = ClientSupport._create_lambda_argument(value)
except Exception:
pytest.fail(f"value of type {type(value)} should be supported")
# assert arg.is_tensor(), "should have been a tensor"
# assert np.all(np.equal(arg.get_tensor_shape(), value.shape))
# assert np.all(
# np.equal(
# value,
# np.array(arg.get_tensor_data()).reshape(arg.get_tensor_shape()),
# )
# )
assert arg.is_tensor(), "should have been a tensor"
assert np.all(np.equal(arg.get_tensor_shape(), value.shape))
assert np.all(
np.equal(
value,
np.array(arg.get_tensor_data()).reshape(arg.get_tensor_shape()),
)
)
def test_accepted_array_as_scalar():