about summary refs log tree commit diff
diff options
context:
space:
mode:
authorArun Isaac2025-09-04 22:05:31 +0100
committerArun Isaac2025-09-04 22:05:31 +0100
commit09da0ae07115600010f267d3c5bd164e7cd786e2 (patch)
tree682cfda2f39399c40277e85e4042551051daf767
parentd1fc4f06b319e94d044fa47f210b2e9628602fdf (diff)
downloadpyhegp-09da0ae07115600010f267d3c5bd164e7cd786e2.tar.gz
pyhegp-09da0ae07115600010f267d3c5bd164e7cd786e2.tar.lz
pyhegp-09da0ae07115600010f267d3c5bd164e7cd786e2.zip
Limit values in genotype and phenotype strategies.
-rw-r--r--tests/helpers/strategies.py8
1 files changed, 6 insertions, 2 deletions
diff --git a/tests/helpers/strategies.py b/tests/helpers/strategies.py
index 50f8049..979ecd7 100644
--- a/tests/helpers/strategies.py
+++ b/tests/helpers/strategies.py
@@ -74,7 +74,9 @@ def genotype_frames(draw,
                                          max_size=_number_of_samples,
                                          unique=True)),
                            dtype="float64",
-                           elements=st.floats(allow_nan=False)))))
+                           elements=st.floats(min_value=0,
+                                              max_value=100,
+                                              allow_nan=False)))))
     return genotype.drop_duplicates(subset=list(
         filter(is_genotype_metadata_column,
                genotype.columns)),
@@ -96,7 +98,9 @@ def phenotype_frames(draw,
                          unique=True)]
                  + columns(draw(phenotype_names),
                            dtype="float64",
-                           elements=st.floats(allow_nan=False))),
+                           elements=st.floats(min_value=-1000,
+                                              max_value=1000,
+                                              allow_nan=False))),
         index=range_indexes(min_size=_number_of_samples,
                             max_size=_number_of_samples)))