diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_pyhegp.py | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/tests/test_pyhegp.py b/tests/test_pyhegp.py index 2d2a258..c494d30 100644 --- a/tests/test_pyhegp.py +++ b/tests/test_pyhegp.py @@ -53,17 +53,11 @@ def no_column_zero_standard_deviation(matrix): elements=st.integers(min_value=0, max_value=100)), arrays("float64", array_shapes(min_dims=2, max_dims=2, min_side=2), - elements=st.floats(min_value=0, max_value=100))) - # Reject matrices with zero standard deviation columns since - # they trigger a division by zero. - .filter(no_column_zero_standard_deviation)) + elements=st.floats(min_value=0, max_value=100)))) def test_hegp_encryption_decryption_are_inverses(plaintext): - mean = np.mean(plaintext, axis=0) - standard_deviation = np.std(plaintext, axis=0) rng = np.random.default_rng() key = random_key(rng, len(plaintext)) - assert hegp_decrypt(hegp_encrypt(plaintext, mean, standard_deviation, key), - mean, standard_deviation, key) == approx(plaintext) + assert hegp_decrypt(hegp_encrypt(plaintext, key), key) == approx(plaintext) @given(arrays("float64", array_shapes(min_dims=2, max_dims=2), |