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: call getChecked to better handle failure

Browse Source
This commit is contained in:
youben11
2021-10-12 15:45:13 +01:00
committed by Quentin Bourgerie
parent 33d75a92f4
commit 6204f93878
4 changed files with 30 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
compiler/tests/python/test_hlfhe_dialect.py
View File

@@ -4,16 +4,17 @@ from zamalang import register_dialects
from zamalang.dialects import hlfhe
def test_eint():
@pytest.mark.parametrize("width", list(range(1, 8)))
def test_eint(width):
ctx = Context()
register_dialects(ctx)
eint = hlfhe.EncryptedIntegerType.get(ctx, 6)
assert eint.__str__() == "!HLFHE.eint<6>"
eint = hlfhe.EncryptedIntegerType.get(ctx, width)
assert eint.__str__() == f"!HLFHE.eint<{width}>"
# FIXME: need to handle error on call to hlfhe.EncryptedIntegerType.get and throw an exception to python
# def test_invalid_eint():
# ctx = Context()
# register_dialects(ctx)
# with pytest.raises(RuntimeError, match=r"mlir parsing failed"):
# eint = hlfhe.EncryptedIntegerType.get(ctx, 16)
@pytest.mark.parametrize("width", [0, 8, 10, 12])
def test_invalid_eint(width):
ctx = Context()
register_dialects(ctx)
with pytest.raises(ValueError, match=r"can't create eint with the given width"):
eint = hlfhe.EncryptedIntegerType.get(ctx, width)