diff options
| author | Arun Isaac | 2025-09-04 20:55:33 +0100 |
|---|---|---|
| committer | Arun Isaac | 2025-09-04 20:55:33 +0100 |
| commit | afca032af0a70b98457ef03f7c218efe2c8d417c (patch) | |
| tree | 2b525b29b9ad2bef6af142dbed284c199a715399 /tests | |
| parent | 5b97a40f493dfd250831d303f5e3e4ba0a8a2c08 (diff) | |
| download | pyhegp-afca032af0a70b98457ef03f7c218efe2c8d417c.tar.gz pyhegp-afca032af0a70b98457ef03f7c218efe2c8d417c.tar.lz pyhegp-afca032af0a70b98457ef03f7c218efe2c8d417c.zip | |
Parameterize presence of reference column in genotype frame strategy.
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/helpers/strategies.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tests/helpers/strategies.py b/tests/helpers/strategies.py index ae35d77..ed9adde 100644 --- a/tests/helpers/strategies.py +++ b/tests/helpers/strategies.py @@ -60,11 +60,13 @@ def summaries(draw): elements=st.floats(allow_nan=False)))))) @st.composite -def genotype_frames(draw): +def genotype_frames(draw, + reference_present=st.booleans()): genotype = draw(data_frames( columns=([chromosome_column, position_column] - + ([reference_column] if draw(st.booleans()) else []) - + columns(draw(sample_names), + + ([reference_column] + if draw(reference_present) + else []) dtype="float64", elements=st.floats(allow_nan=False))))) return genotype.drop_duplicates(subset=list( |