mirror of
https://github.com/zama-ai/concrete.git
synced 2026-02-09 03:55:04 -05:00
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
"""Test file for compilation artifacts"""
|
|
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
from concrete.common.compilation import CompilationArtifacts
|
|
from concrete.common.data_types.integers import UnsignedInteger
|
|
from concrete.common.values import EncryptedScalar
|
|
from concrete.numpy.compile import compile_numpy_function
|
|
|
|
|
|
def test_artifacts_export():
|
|
"""Test function to check exporting compilation artifacts"""
|
|
|
|
def function(x):
|
|
return x + 42
|
|
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
output_directory = Path(tmp)
|
|
artifacts = CompilationArtifacts(output_directory)
|
|
|
|
compile_numpy_function(
|
|
function,
|
|
{"x": EncryptedScalar(UnsignedInteger(7))},
|
|
[(i,) for i in range(10)],
|
|
compilation_artifacts=artifacts,
|
|
)
|
|
|
|
artifacts.export()
|
|
|
|
assert output_directory.joinpath("environment.txt").exists()
|
|
assert output_directory.joinpath("requirements.txt").exists()
|
|
|
|
assert output_directory.joinpath("function.txt").exists()
|
|
assert output_directory.joinpath("parameters.txt").exists()
|
|
|
|
assert output_directory.joinpath("1.initial.graph.txt").exists()
|
|
assert output_directory.joinpath("1.initial.graph.png").exists()
|
|
|
|
assert output_directory.joinpath("2.final.graph.txt").exists()
|
|
assert output_directory.joinpath("2.final.graph.png").exists()
|
|
|
|
assert output_directory.joinpath("bounds.txt").exists()
|
|
assert output_directory.joinpath("mlir.txt").exists()
|
|
|
|
# format of those files might change in the future
|
|
# so it is sufficient to test their existance
|