Skip to content

Commit cb7044f

Browse files
authored
Merge branch 'master' into gh-766-handle-missing-terms-of-use
2 parents 5ed1bb0 + e51965a commit cb7044f

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

malariagen_data/anoph/snp_data.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1872,9 +1872,10 @@ def biallelic_snp_calls(
18721872

18731873
# Apply missingness condition.
18741874
if max_missing_an is not None:
1875-
an_missing = (ds_out.sizes["samples"] * ds_out.sizes["ploidy"]) - an
1875+
an_total = ds_out.sizes["samples"] * ds_out.sizes["ploidy"]
1876+
an_missing = an_total - an
18761877
if isinstance(max_missing_an, float):
1877-
an_missing_frac = an_missing / an
1878+
an_missing_frac = an_missing / an_total
18781879
loc_missing = an_missing_frac <= max_missing_an
18791880
else:
18801881
loc_missing = an_missing <= max_missing_an

tests/anoph/test_snp_data.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1723,8 +1723,9 @@ def test_biallelic_snp_calls_and_diplotypes_with_conditions_fractional(
17231723
an = ac.sum(axis=1)
17241724
ac_min = ac.min(axis=1)
17251725
assert np.all((ac_min / an) >= min_minor_ac)
1726-
an_missing = (ds.sizes["samples"] * ds.sizes["ploidy"]) - an
1727-
assert np.all((an_missing / an) <= max_missing_an)
1726+
an_total = ds.sizes["samples"] * ds.sizes["ploidy"]
1727+
an_missing = an_total - an
1728+
assert np.all((an_missing / an_total) <= max_missing_an)
17281729
gt = ds["call_genotype"].values
17291730
ac_check = allel.GenotypeArray(gt).count_alleles(max_allele=1)
17301731
assert np.all(ac == ac_check)

0 commit comments

Comments
 (0)