def test_write_tree(self):
builder = TreeBuilder()
tree_filename = os.path.join(test_data(), 'animals.terminal_nodes.newick')
tree = Bio.Phylo.read(tree_filename, 'newick')
output = StringIO()
builder._write_tree(tree, output)
with open(tree_filename, 'r') as tree_file:
expected = tree_file.read()
output.seek(0)
self.assertEqual(output.read(), expected)
def test_write_tree(self):
builder = TreeBuilder()
tree_filename = os.path.join(test_data(),
'animals.terminal_nodes.newick')
tree = Bio.Phylo.read(tree_filename, 'newick')
output = StringIO()
builder._write_tree(tree, output)
with open(tree_filename, 'r') as tree_file:
expected = tree_file.read()
output.seek(0)
self.assertEqual(output.read(), expected)
def test_write_output(self):
builder = TreeBuilder()
builder.sequences = {"seq1": "a sequence object"}
builder.tree = "Tree"
builder.sequences_output_file = StringIO()
builder.tree_output_file = StringIO()
builder._write_tree = MagicMock()
builder._write_sequences = MagicMock()
builder.write_output()
builder._write_tree.assert_called_with("Tree",
builder.tree_output_file)
builder._write_sequences.assert_called_with(
["a sequence object"], builder.sequences_output_file)
def test_write_output(self):
builder = TreeBuilder()
builder.sequences = {"seq1": "a sequence object"}
builder.tree = "Tree"
builder.sequences_output_file = StringIO()
builder.tree_output_file = StringIO()
builder._write_tree = MagicMock()
builder._write_sequences = MagicMock()
builder.write_output()
builder._write_tree.assert_called_with("Tree",
builder.tree_output_file)
builder._write_sequences.assert_called_with(["a sequence object"],
builder.sequences_output_file)
