Skip to content

Commit

Permalink
chore: add input compressino factor in test
Browse files Browse the repository at this point in the history
  • Loading branch information
RomanBredehoft committed Apr 29, 2024
1 parent 19d62c8 commit c8825fc
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion tests/deployment/test_client_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -331,7 +331,10 @@ def check_input_compression(model, fhe_circuit_compressed, is_torch, **compilati

uncompressed_size = fhe_circuit_uncompressed.size_of_inputs

assert compressed_size < uncompressed_size, (
# Make sure inputs are compressed by a given important factor
input_compression_factor = 50

assert input_compression_factor * compressed_size < uncompressed_size, (
"Compressed input ciphertext's is not smaller than the uncompressed input ciphertext. Got "
f"{compressed_size} bytes (compressed) and {uncompressed_size} bytes (uncompressed)."
)

0 comments on commit c8825fc

Please sign in to comment.