diff options
author | Arun Isaac | 2025-09-04 20:42:47 +0100 |
---|---|---|
committer | Arun Isaac | 2025-09-04 20:42:47 +0100 |
commit | 5b97a40f493dfd250831d303f5e3e4ba0a8a2c08 (patch) | |
tree | 1354345af1f3a9414f3eb84bbc24d88896c16711 /tests/test_serialization.py | |
parent | 5894ac377c04b4af187309b132c132ff216bb571 (diff) | |
download | pyhegp-5b97a40f493dfd250831d303f5e3e4ba0a8a2c08.tar.gz pyhegp-5b97a40f493dfd250831d303f5e3e4ba0a8a2c08.tar.lz pyhegp-5b97a40f493dfd250831d303f5e3e4ba0a8a2c08.zip |
Add keys strategy.
Add keys strategy, and use it.
Diffstat (limited to 'tests/test_serialization.py')
-rw-r--r-- | tests/test_serialization.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tests/test_serialization.py b/tests/test_serialization.py index ab354e1..32cb58f 100644 --- a/tests/test_serialization.py +++ b/tests/test_serialization.py @@ -80,8 +80,7 @@ def test_read_write_phenotype_are_inverses(phenotype): file.seek(0) pd.testing.assert_frame_equal(phenotype, read_phenotype(file)) -@given(arrays("float64", - array_shapes(min_dims=2, max_dims=2))) +@given(keys(st.integers(min_value=2, max_value=10))) def test_read_write_key_are_inverses(key): with tempfile.TemporaryFile() as file: write_key(file, key) |