t/Tree/TreeIO/nhx.t

   1 # -*-Perl-*- Test Harness script for Bioperl
   2 # $Id: TreeIO.t 14580 2008-03-01 17:01:30Z cjfields $
   3
   4 use strict;
   5
   6 BEGIN {
   7     use lib '.';
   8     use Bio::Root::Test;
   9     use File::Temp qw(tempfile);
  10
  11     test_begin(-tests => 19);
  12     use_ok('Bio::TreeIO');
  13 }
  14
  15 my $verbose = 0; #test_debug();
  16 my $nl = qr/\n/;
  17 my $cr = qr/\r/;
  18
  19 my $treeio = Bio::TreeIO->new(
  20   -format => 'nhx',
  21   -verbose => $verbose,
  22   -file   => test_input_file('test.nhx'),
  23   );
  24 my $tree;
  25 ok($treeio);
  26 $tree = $treeio->next_tree;
  27 isa_ok($tree, 'Bio::Tree::TreeI');
  28
  29 my @nodes = $tree->get_nodes;
  30 is(@nodes, 12, "Total Nodes");
  31 #print STDERR "TREE: ".$tree->as_text('nhx')."\n";
  32
  33 my $adhy = $tree->find_node('ADHY');
  34 is($adhy->branch_length, 0.1);
  35 is(($adhy->get_tag_values('S'))[0], 'nematode');
  36 is(($adhy->get_tag_values('E'))[0], '1.1.1.1');
  37
  38 test_roundtrip('((a,b),c);','simple newick');
  39 test_roundtrip('((x:0.05,y:0.06),a:0.1[&&NHX:G=dummy]);','bug 1471 test');
  40 test_roundtrip('((x:0.05[&&NHX:label=x],y:0.06)[&&NHX:label=int_node],a:0.1[&&NHX:label=a]);','different combinations of label, NHX, and branch length');
  41
  42 test_roundtrip('(a:1,b:2,c:3,d:4)TEST:1.2345;','doot node branch length');
  43 test_roundtrip('(A:0.1,B:0.2,(C:0.3,D:0.4)E:0.5)F;','Example from Wikipedia');
  44
  45 test_roundtrip('(((ADH2:0.1[&&NHX:E=1.1.1.1:S=human],ADH1:0.11[&&NHX:E=1.1.1.1:S=human]):0.05[&&NHX:B=100:D=Y:E=1.1.1.1:S=Primates],ADHY:0.1[&&NHX:E=1.1.1.1:S=nematode],ADHX:0.12[&&NHX:E=1.1.1.1:S=insect]):0.1[&&NHX:D=N:E=1.1.1.1:S=Metazoa],(ADH4:0.09[&&NHX:E=1.1.1.1:S=yeast],ADH3:0.13[&&NHX:E=1.1.1.1:S=yeast],ADH2:0.12[&&NHX:E=1.1.1.1:S=yeast],ADH1:0.11[&&NHX:E=1.1.1.1:S=yeast]):0.1[&&NHX:S=Fungi])[&&NHX:D=N:E=1.1.1.1];','ADH NHX tree');
  46 test_roundtrip('(gene1_Hu[&&NHX:S=Hu_Homo_sapiens],(gene2_Hu[&&NHX:S=Hu_Homo_sapiens],gene2_Mu[&&NHX:S=Mu_Mus_musculus]));','notung nhx example http://www.cs.cmu.edu/~aiton/split/Manual-2.6.master014.html');
  47 test_roundtrip('(cow_gene1,(mouse_gene2,cow_gene2)[&&NHX:B=100]);','notung nhx bootstrap http://www.cs.cmu.edu/~aiton/split/Manual-2.6.master014.html');
  48
  49 # Read in some larger trees from data files...
  50 test_roundtrip(read_file(test_input_file('nhx-bacteria.nhx')),'r-sig-phylo mailing list http://www.mail-archive.com/r-sig-phylo@r-project.org/msg00516.html');
  51 test_roundtrip(read_file(test_input_file('ex1.nucl.nhx')),'treebest example nhx');
  52 # Note: these files aren't reproduced exactly in their online form. We need to round-trip them once
  53 # before including them in the test, because the ordering of annotation keys is not a well-defined
  54 # part of the NHX format. Since nhx.pm sorts the keys before output, once they've been through
  55 # one time, the ordering becomes stable.
  56 test_roundtrip(read_file(test_input_file('wellcome_tol.nhx')),'Wellcome Trust ToL (from http://iphylo.blogspot.com/2009/02/thoughts-on-wellcome-interactive-tree.html)');
  57
  58 # Uncomment to run (takes a long time!!)
  59 #test_roundtrip(read_file(test_input_file('tol-2010-02-18.nhx')),'Tolweb.org converted to NHX');
  60
  61 test_roundtrip(read_file(test_input_file('biorecipe.nhx')),'Biorecipes NHX file (http://www.biorecipes.com/Orthologues/StatusPage/pics/TreeEukaryota.nt)');
  62
  63 sub test_roundtrip {
  64   my $string = shift;
  65   my $desc = shift;
  66
  67   my $in = Bio::TreeIO->new(-format => 'nhx',
  68                             -string => $string,
  69                             -verbose => $verbose
  70                             );
  71
  72   my $t = $in->next_tree;
  73   my $out;
  74   if (defined $t) {
  75     $out = $t->as_text('nhx');
  76   }
  77
  78   $desc = "Roundtrip: $desc";
  79   return is($out,$string,$desc);
  80 }
  81
  82 sub read_file {
  83   my $file = shift;
  84   local $/=undef;
  85   my $string;
  86   open my $IN, '<', $file or die "Could not read file '$file': $!\n";
  87   binmode $IN;
  88   $string = <$IN>;
  89   close $IN;
  90   $string =~ s/$nl//g;
  91   $string =~ s/$cr//g; # For files with Windows line-endings
  92   #print STDERR "STR: $string\n";
  93   return $string;
  94 }