about summary refs log tree commit diff
path: root/tests/test_serialization.py
diff options
context:
space:
mode:
authorArun Isaac2025-09-04 20:42:47 +0100
committerArun Isaac2025-09-04 20:42:47 +0100
commit5b97a40f493dfd250831d303f5e3e4ba0a8a2c08 (patch)
tree1354345af1f3a9414f3eb84bbc24d88896c16711 /tests/test_serialization.py
parent5894ac377c04b4af187309b132c132ff216bb571 (diff)
downloadpyhegp-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.py3
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)