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)