def test_consistency_unrooted(self):
"""Test consistency of taxa with a taxa that is only monophyletic in unrooted tree"""
seed_con = 'f__Lachnospiraceae; g__Bacteroides; s__'
nl.determine_rank_order(seed_con)
tipname_map = {'a': ['f__Lachnospiraceae', 'g__Bacteroides', 's__Bacteroides pectinophilus'],
'b': ['f__Lachnospiraceae', 'g__Bacteroides', 's__Bacteroides pectinophilus'],
'c': ['f__Lachnospiraceae', 'g__Bacteroides', 's__Bacteroides pectinophilus'],
'd': ['f__Lachnospiraceae', 'g__Bacteroides', 's__Bacteroides acidifaciens'],
'e': ['f__Lachnospiraceae', 'g__Bacteroides', 's__Bacteroides acidifaciens']}
tree = nl.load_tree(StringIO(u'((a,b),(c,(d,e)));'), tipname_map)
counts = nl.collect_names_at_ranks_counts(tree)
nl.decorate_ntips_rank(tree)
nl.decorate_name_counts(tree)
# determine taxonomic consistency of rooted tree
#expected_consistency_index
c = Consistency(counts, len(nl.RANK_ORDER))
consistency_index = c.calculate(tree, rooted=True)
self.assertAlmostEqual(consistency_index[0]['f__Lachnospiraceae'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Bacteroides'], 1.0)
self.assertAlmostEqual(consistency_index[2]['s__Bacteroides pectinophilus'], 0.66666666)
self.assertAlmostEqual(consistency_index[2]['s__Bacteroides acidifaciens'], 1.0)
#determine consistency of unrooted tree
consistency_index = c.calculate(tree, rooted=False)
self.assertAlmostEqual(consistency_index[0]['f__Lachnospiraceae'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Bacteroides'], 1.0)
self.assertAlmostEqual(consistency_index[2]['s__Bacteroides pectinophilus'], 1.0)
self.assertAlmostEqual(consistency_index[2]['s__Bacteroides acidifaciens'], 1.0)
def test_consistency_missing(self):
"""Test consistency of taxa in tree with missing taxa"""
seed_con = 'f__Lachnospiraceae; g__Bacteroides; s__'
nl.determine_rank_order(seed_con)
tipname_map = {'a': ['f__Lachnospiraceae', 'g__Bacteroides', None],
'c': ['f__Lachnospiraceae', 'g__Bacteroides', 's__Bacteroides pectinophilus'],
'b': ['f__Lachnospiraceae', 'g__Bacteroides', None], 'e': [None, None, None],
'd': ['f__Lachnospiraceae', 'g__Bacteroides', 's__Bacteroides pectinophilus'],
'g': [None, None, None], 'f': ['f__Lachnospiraceae', 'g__Lachnospira', None],
'h': ['f__Lachnospiraceae', 'g__Lachnospira', 's__Bacteroides pectinophilus']}
tree = nl.load_tree(StringIO(u'(((a,b),(c,d)),((e,f),(g,h)));'), tipname_map)
counts = nl.collect_names_at_ranks_counts(tree)
nl.decorate_ntips_rank(tree)
nl.decorate_name_counts(tree)
# determine taxonomic consistency of rooted tree
#expected_consistency_index
c = Consistency(counts, len(nl.RANK_ORDER))
consistency_index = c.calculate(tree, rooted=True)
self.assertAlmostEqual(consistency_index[0]['f__Lachnospiraceae'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Bacteroides'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Lachnospira'], 1.0)
self.assertAlmostEqual(consistency_index[2]['s__Bacteroides pectinophilus'], 1.0)
#determine consistency of unrooted tree
consistency_index = c.calculate(tree, rooted=False)
self.assertAlmostEqual(consistency_index[0]['f__Lachnospiraceae'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Bacteroides'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Lachnospira'], 1.0)
self.assertAlmostEqual(consistency_index[2]['s__Bacteroides pectinophilus'], 1.0)
def test_consistency_unrooted(self):
"""Test consistency of taxa with a taxa that is only monophyletic in unrooted tree"""
seed_con = 'f__Lachnospiraceae; g__Bacteroides; s__'
nl.determine_rank_order(seed_con)
tipname_map = {
'a': [
'f__Lachnospiraceae', 'g__Bacteroides',
's__Bacteroides pectinophilus'
],
'b': [
'f__Lachnospiraceae', 'g__Bacteroides',
's__Bacteroides pectinophilus'
],
'c': [
'f__Lachnospiraceae', 'g__Bacteroides',
's__Bacteroides pectinophilus'
],
'd': [
'f__Lachnospiraceae', 'g__Bacteroides',
's__Bacteroides acidifaciens'
],
'e': [
'f__Lachnospiraceae', 'g__Bacteroides',
's__Bacteroides acidifaciens'
]
}
tree = nl.load_tree('((a,b),(c,(d,e)));', tipname_map)
counts = nl.collect_names_at_ranks_counts(tree)
nl.decorate_ntips_rank(tree)
nl.decorate_name_counts(tree)
# determine taxonomic consistency of rooted tree
#expected_consistency_index
c = Consistency(counts, len(nl.RANK_ORDER))
consistency_index = c.calculate(tree, rooted=True)
self.assertAlmostEqual(consistency_index[0]['f__Lachnospiraceae'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Bacteroides'], 1.0)
self.assertAlmostEqual(
consistency_index[2]['s__Bacteroides pectinophilus'], 0.66666666)
self.assertAlmostEqual(
consistency_index[2]['s__Bacteroides acidifaciens'], 1.0)
#determine consistency of unrooted tree
consistency_index = c.calculate(tree, rooted=False)
self.assertAlmostEqual(consistency_index[0]['f__Lachnospiraceae'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Bacteroides'], 1.0)
self.assertAlmostEqual(
consistency_index[2]['s__Bacteroides pectinophilus'], 1.0)
self.assertAlmostEqual(
consistency_index[2]['s__Bacteroides acidifaciens'], 1.0)
def test_consistency_missing(self):
"""Test consistency of taxa in tree with missing taxa"""
seed_con = 'f__Lachnospiraceae; g__Bacteroides; s__'
nl.determine_rank_order(seed_con)
tipname_map = {
'a': ['f__Lachnospiraceae', 'g__Bacteroides', None],
'c': [
'f__Lachnospiraceae', 'g__Bacteroides',
's__Bacteroides pectinophilus'
],
'b': ['f__Lachnospiraceae', 'g__Bacteroides', None],
'e': [None, None, None],
'd': [
'f__Lachnospiraceae', 'g__Bacteroides',
's__Bacteroides pectinophilus'
],
'g': [None, None, None],
'f': ['f__Lachnospiraceae', 'g__Lachnospira', None],
'h': [
'f__Lachnospiraceae', 'g__Lachnospira',
's__Bacteroides pectinophilus'
]
}
tree = nl.load_tree('(((a,b),(c,d)),((e,f),(g,h)));', tipname_map)
counts = nl.collect_names_at_ranks_counts(tree)
nl.decorate_ntips_rank(tree)
nl.decorate_name_counts(tree)
# determine taxonomic consistency of rooted tree
#expected_consistency_index
c = Consistency(counts, len(nl.RANK_ORDER))
consistency_index = c.calculate(tree, rooted=True)
self.assertAlmostEqual(consistency_index[0]['f__Lachnospiraceae'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Bacteroides'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Lachnospira'], 1.0)
self.assertAlmostEqual(
consistency_index[2]['s__Bacteroides pectinophilus'], 1.0)
#determine consistency of unrooted tree
consistency_index = c.calculate(tree, rooted=False)
self.assertAlmostEqual(consistency_index[0]['f__Lachnospiraceae'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Bacteroides'], 1.0)
self.assertAlmostEqual(consistency_index[1]['g__Lachnospira'], 1.0)
self.assertAlmostEqual(
consistency_index[2]['s__Bacteroides pectinophilus'], 1.0)
def test_decorate_name_counts(self):
"""correctly decorate relative frequency information on a tree"""
input = "((a,b)c,(d,(e,f)g)h,(i,j)k)l;"
tipname_map = {
'a': ['1', '2', '3', '4', '5', '6', '7'],
'b': ['1', '2', '3', '4', '5', '6', '8'],
'd': ['1', '2', '3', '4', '5', '6', '8'],
'e': ['1', '2', '3', '4', 'a', '6', '7'],
'i': ['1', '2', '3', '4', 'a', None, '7'],
'j': ['1', '2', '3', '4', 'a', None, '8']
}
tree = load_tree(input, tipname_map)
decorate_ntips(tree)
decorate_name_counts(tree)
exp_root = {
0: {
'1': 6
},
1: {
'2': 6
},
2: {
'3': 6
},
3: {
'4': 6
},
4: {
'5': 3,
'a': 3
},
5: {
'6': 4
},
6: {
'7': 3,
'8': 3
}
}
self.assertEqual(tree.TaxaCount, exp_root)
def test_decorate_name_counts(self):
"""correctly decorate relative frequency information on a tree"""
input = "((a,b)c,(d,(e,f)g)h,(i,j)k)l;"
tipname_map = {'a': ['1', '2', '3', '4', '5', '6', '7'],
'b': ['1', '2', '3', '4', '5', '6', '8'],
'd': ['1', '2', '3', '4', '5', '6', '8'],
'e': ['1', '2', '3', '4', 'a', '6', '7'],
'i': ['1', '2', '3', '4', 'a', None, '7'],
'j': ['1', '2', '3', '4', 'a', None, '8']}
tree = load_tree(input, tipname_map)
decorate_ntips(tree)
decorate_name_counts(tree)
exp_root = {0: {'1': 6},
1: {'2': 6},
2: {'3': 6},
3: {'4': 6},
4: {'5': 3, 'a': 3},
5: {'6': 4},
6: {'7': 3, '8': 3}}
self.assertEqual(tree.TaxaCount, exp_root)
