def test_tree_demo_normalization():
lins_per_pop = 10
num_leaf_pops = 3
demo = random_tree_demo(num_leaf_pops, lins_per_pop)
check_demo_normalization(
demo._get_demo(dict(zip(demo.leafs, [lins_per_pop] * num_leaf_pops))))
def test_tree_demo_numsnps():
lins_per_pop = 10
num_leaf_pops = 3
demo = random_tree_demo(num_leaf_pops, lins_per_pop)
check_num_snps(dict(zip(demo.leafs, [10] * num_leaf_pops)), demo, 1000.0,
1.0)
def test_tree_demo_rank1tensor():
lins_per_pop = 2
num_leaf_pops = 3
demo = random_tree_demo(num_leaf_pops, lins_per_pop)
check_random_tensor(
demo._get_demo(dict(zip(demo.leafs, [lins_per_pop] * num_leaf_pops))))