diff options
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) |