X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2Fstemma.t;h=404b686bbeab8a9368c0fd1e9fbc0aaf6b1696bb;hb=c84a47788777f257a330f9d011c03077e622310e;hp=276e7366fa443dae5239ed5d15bd5a6a45ca5fe0;hpb=7035e3a6ff8829ec8390fdfde75f23def2ed9313;p=scpubgit%2Fstemmatology.git

diff --git a/t/stemma.t b/t/stemma.t
index 276e736..404b686 100644
--- a/t/stemma.t
+++ b/t/stemma.t
@@ -1,10 +1,11 @@
 #!/usr/bin/perl
 
 use strict; use warnings;
+use File::Which;
 use Test::More;
 use lib 'lib';
 use Text::Tradition;
-use Text::Tradition::Stemma;
+use Text::Tradition::StemmaUtil qw/ character_input phylip_pars parse_newick /;
 use XML::LibXML;
 use XML::LibXML::XPathContext;
 
@@ -15,23 +16,61 @@ my $tradition = Text::Tradition->new(
     'input' => 'CollateX',
     'file'  => $datafile,
     );
-my $stemma = Text::Tradition::Stemma->new( 'collation' => $tradition->collation );
+# Set up some relationships
+my $c = $tradition->collation;
+$c->add_relationship( 'n23', 'n24', { 'type' => 'spelling' } );
+$c->add_relationship( 'n9', 'n10', { 'type' => 'spelling' } );
+$c->add_relationship( 'n12', 'n13', { 'type' => 'spelling' } );
+$c->calculate_ranks();
+
+my $stemma = $tradition->add_stemma( dotfile => 't/data/simple.dot' );
 
 # Test for object creation
 ok( $stemma->isa( 'Text::Tradition::Stemma' ), 'Got the right sort of object' );
+is( $stemma->graph, '1-2,1-A,2-B,2-C', "Got the correct graph" );
 
 # Test for character matrix creation
-$stemma->make_character_matrix();
+my $mstr = character_input( $c->alignment_table() );
  ## check number of rows
+my @mlines = split( "\n", $mstr );
+my $msig = shift @mlines;
+my( $rows, $chars ) = $msig =~ /(\d+)\s+(\d+)/;
+is( $rows, 3, "Found three witnesses in char matrix" );
  ## check number of columns
+is( $chars, 18, "Found 18 rows plus sigla in char matrix" );
+ ## check matrix
+my %expected = (
+	'A' => 'AAAAAAAXAAAAAAAAAA',
+	'B' => 'AXXXAAAAAABABAABAA',
+	'C' => 'AXXXAAAAABAAAAAXBB',
+	);
+foreach my $ml ( @mlines ) {
+	my( $wit, $chars ) = split( /\s+/, $ml );
+	is( $chars, $expected{$wit}, "Row for witness $wit is correct" );
+}
 
 # Test that pars runs
-my( $status, $tree ) = $stemma->run_phylip_pars();
-ok( $status, "pars ran successfully" );
-print STDERR "Error was $tree\n" unless $status;
+SKIP: {
+    skip "pars not in path", 3 unless File::Which::which('pars');
+    my $newick = phylip_pars( $mstr );
+    ok( $newick, "pars ran successfully" );
 
-# Test that we get a tree
+	my $trees = parse_newick( $newick );
+    # Test that we get a tree
+    is( scalar @$trees, 1, "Got a single tree" );
+    # Test that the tree has all our witnesses
+    my $tree = $trees->[0];
+    my @leaves = grep { $tree->degree( $_ ) == 1 } $tree->vertices;
+    is( scalar @leaves, 3, "All witnesses in the tree" );
+}
 
-# Test that the tree has all our witnesses
+# Test our dot output
+my $display = $stemma->as_dot();
+ok( $display =~ /digraph/, "Got a dot display graph" );
+ok( $display !~ /hypothetical/, "Graph is display rather than edit" );
+# Test our editable output
+my $editable = $stemma->editable();
+ok( $editable =~ /digraph/, "Got a dot edit graph" );
+ok( $editable =~ /hypothetical/, "Graph contains an edit class" );
 
-done_testing();
\ No newline at end of file
+done_testing();