/
test_Phylo_NeXML.py
127 lines (101 loc) · 3.91 KB
/
test_Phylo_NeXML.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
# Copyright (C) 2013 by Ben Morris (ben@bendmorris.com)
# based on code by Eric Talevich (eric.talevich@gmail.com)
# This code is part of the Biopython distribution and governed by its
# license. Please see the LICENSE file that should have been included
# as part of this package.
"""Unit tests for the NeXML and NeXMLIO modules.
"""
import os
import tempfile
import unittest
import Bio.Phylo as bp
from Bio.Phylo import NeXML, NeXMLIO
# Example NeXML files
nexml_files = (
'characters.xml',
'edgelabels.xml',
'meta_taxa.xml',
'meta_types.xml',
'nexml.xml',
'phenoscape.xml',
'sets.xml',
'taxa.xml',
'timetree.xml',
'tolweb.xml',
'treebase-record.xml',
'trees-uris.xml',
'trees.xml',
)
tree_counts = {
'taxa.xml': 0,
'timetree.xml': 38,
'phenoscape.xml': 0,
'nexml.xml': 0,
'meta_types.xml': 0,
'meta_taxa.xml': 0,
'trees.xml': 2,
'characters.xml': 0,
}
# Temporary file name for Writer tests below
DUMMY = tempfile.mktemp()
# ---------------------------------------------------------
# Parser tests
def _test_parse_factory(source):
"""Generate a test method for parse()ing the given source.
The generated function extracts each phylogenetic tree using the parse()
function and counts the total number of trees extracted.
"""
filename = os.path.join('NeXML/', source)
if source in tree_counts: count = tree_counts[source]
else: count = 1
def test_parse(self):
trees = list(bp._io.parse(filename, 'nexml'))
self.assertEqual(len(trees), count)
test_parse.__doc__ = "Parse the phylogenies in %s." % source
return test_parse
def _test_write_factory(source):
"""Tests for serialization of objects to NeXML format.
Modifies the globally defined filenames in order to run the other parser
tests on files (re)generated by NeXMLIO's own writer.
"""
filename = os.path.join('NeXML/', source)
if source in tree_counts: count = tree_counts[source]
else: count = 1
def test_write(self):
"""Parse, rewrite and retest an example file."""
infile = open(filename, 'rb')
t1 = NeXMLIO.Parser(infile).parse().next()
infile.close()
outfile = open(DUMMY, 'w+b')
NeXMLIO.write([t1], outfile)
outfile.close()
t2 = NeXMLIO.Parser(open(DUMMY, 'rb')).parse().next()
def assert_property(prop_name):
p1 = sorted([getattr(n, prop_name) for n in t1.get_terminals() if getattr(n, prop_name)])
p2 = sorted([getattr(n, prop_name) for n in t2.get_terminals() if getattr(n, prop_name)])
self.assertEqual(p1,p2)
for prop_name in ('name', 'branch_length', 'confidence'):
assert_property(prop_name)
test_write.__doc__ = "Write and re-parse the phylogenies in %s." % source
return test_write
class ParseTests(unittest.TestCase):
"""Tests for proper parsing of example NeXML files."""
for n, ex in enumerate(nexml_files):
parse_test = _test_parse_factory(ex)
parse_test.__name__ = 'test_parse_%s' % n
setattr(ParseTests, parse_test.__name__, parse_test)
class WriterTests(unittest.TestCase):
pass
for n, ex in enumerate(nexml_files):
count = 1
if ex in tree_counts: count = tree_counts[ex]
if count > 0:
write_test = _test_write_factory(ex)
write_test.__name__ = 'test_write_%s' % n
setattr(WriterTests, write_test.__name__, write_test)
if __name__ == '__main__':
runner = unittest.TextTestRunner(verbosity=2)
unittest.main(testRunner=runner)
# Clean up the temporary file
if os.path.exists(DUMMY):
os.remove(DUMMY)