diff options
| -rw-r--r-- | tests/test_pyhegp.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tests/test_pyhegp.py b/tests/test_pyhegp.py index 567ee17..61358e5 100644 --- a/tests/test_pyhegp.py +++ b/tests/test_pyhegp.py @@ -25,18 +25,17 @@ from pyhegp.pyhegp import hegp_encrypt, hegp_decrypt, random_key @given(st.one_of( arrays("int32", - array_shapes(min_dims=2, max_dims=2, min_side=2, max_side=100), + array_shapes(min_dims=2, max_dims=2), elements=st.integers(min_value=0, max_value=2)), # The array above is the only realistic input, but we test more # kinds of inputs for good measure. arrays("int32", - array_shapes(min_dims=2, max_dims=2, min_side=2, max_side=100), + array_shapes(min_dims=2, max_dims=2), elements=st.integers(min_value=0, max_value=100)), arrays("float64", - array_shapes(min_dims=2, max_dims=2, min_side=2, max_side=100), + array_shapes(min_dims=2, max_dims=2), elements=st.floats(min_value=0, max_value=100))) ) -@settings(deadline=None) def test_hegp_encryption_decryption_are_inverses(plaintext): rng = np.random.default_rng() key = random_key(rng, len(plaintext)) |