From f5a245d1dc7e7680a979c52ec77cdc3e54ccea4a Mon Sep 17 00:00:00 2001 From: Arun Isaac Date: Mon, 7 Jul 2025 12:17:28 +0100 Subject: Add tests. * test_pyhegp.py: New file. * README.md (Run tests): New section. * guix.scm: Import python-hypothesis from (gnu packages check). (python-pyhegp)[arguments]: Enable tests. [native-inputs]: Add python-hypothesis. --- test_pyhegp.py | 45 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 45 insertions(+) create mode 100644 test_pyhegp.py (limited to 'test_pyhegp.py') diff --git a/test_pyhegp.py b/test_pyhegp.py new file mode 100644 index 0000000..19e58d2 --- /dev/null +++ b/test_pyhegp.py @@ -0,0 +1,45 @@ +### pyhegp --- Homomorphic encryption of genotypes and phenotypes +### Copyright © 2025 Arun Isaac +### +### This file is part of pyhegp. +### +### pyhegp is free software: you can redistribute it and/or modify it +### under the terms of the GNU General Public License as published by +### the Free Software Foundation, either version 3 of the License, or +### (at your option) any later version. +### +### pyhegp is distributed in the hope that it will be useful, but +### WITHOUT ANY WARRANTY; without even the implied warranty of +### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +### General Public License for more details. +### +### You should have received a copy of the GNU General Public License +### along with pyhegp. If not, see . + +from hypothesis import given, settings, strategies as st +from hypothesis.extra.numpy import arrays, array_shapes +import numpy as np +from pytest import approx + +from 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=200), + 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=200), + elements=st.integers(min_value=0, max_value=100)), + arrays("float64", + array_shapes(min_dims=2, max_dims=2, min_side=2, max_side=200), + 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)) + # FIXME: We don't use maf at the moment. + maf = None + assert hegp_decrypt(hegp_encrypt(plaintext, maf, key), key) == approx(plaintext) -- cgit v1.2.3