Get rid of Graph::Easy; add stemma tests
[scpubgit/stemmatology.git] / t / stemma.t
1 #!/usr/bin/perl
2
3 use strict; use warnings;
4 use Test::More;
5 use lib 'lib';
6 use Text::Tradition;
7 use XML::LibXML;
8 use XML::LibXML::XPathContext;
9
10 my $datafile = 't/data/Collatex-16.xml'; #TODO need other test data
11
12 my $tradition = Text::Tradition->new( 
13     'name'  => 'inline', 
14     'input' => 'CollateX',
15     'file'  => $datafile,
16     );
17 # Set up some relationships
18 my $c = $tradition->collation;
19 $c->add_relationship( 'n25', 'n26', { 'type' => 'spelling' } );
20 $c->add_relationship( 'n9', 'n23', { 'type' => 'spelling' } );
21 $c->add_relationship( 'n8', 'n13', { 'type' => 'spelling' } );
22 $c->calculate_ranks();
23
24 my $stemma = $tradition->add_stemma( 't/data/simple.dot' );
25
26 # Test for object creation
27 ok( $stemma->isa( 'Text::Tradition::Stemma' ), 'Got the right sort of object' );
28 is( $stemma->graph, '1-2,1-A,2-B,2-C', "Got the correct graph" );
29
30 # Test for character matrix creation
31 my $m = $stemma->make_character_matrix();
32  ## check number of rows
33 is( scalar @$m, 3, "Found three witnesses in char matrix" );
34  ## check number of columns
35 is( scalar( @{$m->[0]} ), 19, "Found 18 rows plus sigla in char matrix" );
36  ## check matrix
37 my %expected = (
38         'A' => 'AAAAAAAXAAAAAAAAAA',
39         'B' => 'AXXXAAAAAABABAABAA',
40         'C' => 'AXXXAAAAABAAAAAXBB',
41         );
42 my @wits = map { shift @$_; } @$m;
43 map { s/\s+//g } @wits;
44 foreach my $i ( 0 .. $#wits ) {
45         my $w = $wits[$i];
46         is( join( '', @{$m->[$i]} ), $expected{$w}, "Row for witness $w is correct" );
47 }
48
49 # Test that pars runs
50 my( $status, $tree ) = $stemma->run_phylip_pars();
51 ok( $status, "pars ran successfully" );
52 print STDERR "Error was $tree\n" unless $status;
53
54 # Test that we get a tree
55 is( scalar @{$stemma->distance_trees}, 1, "Got a single tree" );
56 # Test that the tree has all our witnesses
57 $tree = $stemma->distance_trees->[0];
58 my @leaves = grep { $tree->degree( $_ ) == 1 } $tree->vertices;
59 is( scalar @leaves, 3, "All witnesses in the tree" );
60
61 # Test our dot output
62 my $display = $stemma->as_dot();
63 ok( $display =~ /digraph/, "Got a dot display graph" );
64 ok( $display !~ /hypothetical/, "Graph is display rather than edit" );
65 # Test our editable output
66 my $editable = $stemma->editable();
67 ok( $editable =~ /digraph/, "Got a dot edit graph" );
68 ok( $editable =~ /hypothetical/, "Graph contains an edit class" );
69
70 done_testing();