From 3f3dd13f75ab91862c9e0cbd5e65f1da1e26cf4b Mon Sep 17 00:00:00 2001 From: Arun Isaac Date: Tue, 2 Sep 2025 17:33:08 +0100 Subject: Drop duplicates in generated test genotype frames. --- tests/helpers/strategies.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/helpers/strategies.py b/tests/helpers/strategies.py index fc8ae1d..7edf667 100644 --- a/tests/helpers/strategies.py +++ b/tests/helpers/strategies.py @@ -63,12 +63,16 @@ def summaries(draw): @st.composite def genotype_frames(draw): - return draw(data_frames( + genotype = draw(data_frames( columns=([chromosome_column, position_column] + ([reference_column] if draw(st.booleans()) else []) + columns(draw(sample_names), dtype="float64", elements=st.floats(allow_nan=False))))) + return genotype.drop_duplicates(subset=list( + filter(genotype_reserved_column_name_p, + genotype.columns)), + ignore_index=True) def phenotype_reserved_column_name_p(name): return name.lower() == "sample-id" -- cgit 1.4.1