Skip to content

Commit 3eaaa93

Browse files
authored
Merge pull request #1039 from petrelharp/sample_tests
Test stats when not all samples are included.
2 parents accfac6 + a7c9e66 commit 3eaaa93

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

python/tests/test_tree_stats.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -881,8 +881,12 @@ def example_sample_sets(ts, min_size=1):
881881
yield [samples[:1]]
882882
if ts.num_samples > 2 and min_size <= 2:
883883
yield [samples[:2], samples[2:]]
884+
if ts.num_samples > 4 and min_size <= 2:
885+
yield [samples[:2], samples[2:4]]
884886
if ts.num_samples > 7 and min_size <= 4:
885887
yield [samples[:2], samples[2:4], samples[4:6], samples[6:]]
888+
if ts.num_samples > 8 and min_size <= 4:
889+
yield [samples[:2], samples[2:4], samples[4:6], samples[6:8]]
886890

887891

888892
def example_sample_set_index_pairs(sample_sets):

0 commit comments

Comments
 (0)