From afca032af0a70b98457ef03f7c218efe2c8d417c Mon Sep 17 00:00:00 2001
From: Arun Isaac
Date: Thu, 4 Sep 2025 20:55:33 +0100
Subject: Parameterize presence of reference column in genotype frame strategy.

---
 tests/helpers/strategies.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'tests')

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(
-- 
cgit 1.4.1