def test_admixture_3pop_ascertainment():
check_demo_normalization(simple_admixture_3pop()._get_demo({
"a": 5,
"b": 5,
"c": 5
}),
ascertainment_pop=[True, True, False])
def test_read_vcf():
sampled_n_dict = {"a": 4, "b": 4, "c": 6}
demo = demo_utils.simple_admixture_3pop()
theta = 100.0
rho = 100.0
num_loci = 5
num_bases = 1e5
demo.simulate_vcf("test_vcf",
recoms_per_gen=rho / num_bases,
length=num_bases,
muts_per_gen=theta / num_bases,
sampled_n_dict=sampled_n_dict,
random_seed=1234,
force=True)
data = demo.simulate_data(recoms_per_gen=rho / num_bases,
length=num_bases,
muts_per_gen=theta / num_bases,
sampled_n_dict=sampled_n_dict,
num_replicates=1,
random_seed=1234)
data2 = momi.SnpAlleleCounts.read_vcf(
'test_vcf.vcf.gz',
ind2pop={
f"{pop}_{i}": pop
for pop, n in sampled_n_dict.items() for i in range(n)
})
assert data._sfs == data2._sfs.subset_populations(data._sfs.sampled_pops)
def test_admixture_3pop_numsnps():
check_num_snps({
"a": 1,
"b": 2,
"c": 3
}, simple_admixture_3pop(), 1000.0, 1.0)