1 #include <cppunit/TestFixture.h>
2 #include <cppunit/extensions/HelperMacros.h>
4 #include <genesorter.h>
6 #include <genealgorithms.h>
7 #include <componenttree.h>
14 * A test case that is designed to produce
15 * example errors and failures.
19 #define TESTNAME ComponentTreeTest
22 class TESTNAME : public CPPUNIT_NS::TestFixture
24 CPPUNIT_TEST_SUITE( TESTNAME );
25 CPPUNIT_TEST( testCreate );
26 CPPUNIT_TEST( testMakeUnoriented );
27 CPPUNIT_TEST( testCountLeaves );
28 CPPUNIT_TEST_SUITE_END();
31 vector<int> _validPerm;
32 vector<int> _validPerm2;
33 vector<int> _validPerm3;
34 vector<int> _validPerm4;
39 int validPerm[] = {1,2,3,4};
40 _validPerm.assign(validPerm,validPerm+4);
41 int validPerm2[] = {1,-3,-2,4};
42 _validPerm2.assign(validPerm2,validPerm2+4);
43 int validPerm3[] = {0,-2,-1,4,3,5,-8,6,7,9};
44 _validPerm3.assign(validPerm3,validPerm3+10);
45 int validPerm4[] = {-3,1,2,4,6,5,7,-15,-13,-14,-12,-10,-11,-9,8};
46 _validPerm4.assign(validPerm4,validPerm4+15);
52 size_t count(ComponentTree::Node* n, bool all = false){
54 if (all || n->_comp.i2 != 0)
56 for (vector<ComponentTree::Node*>::iterator c = n->_children.begin();
57 c != n->_children.end(); ++c){
63 GeneOrder go(_validPerm.begin(),_validPerm.end());
64 ComponentTree t(findComponents(go));
65 ComponentTree::Node* n = t._root;
67 CPPUNIT_ASSERT_EQUAL((size_t)4u,count(n));
68 CPPUNIT_ASSERT_EQUAL((size_t)5u,count(n,true));
69 CPPUNIT_ASSERT_EQUAL((size_t)4u,n->_children.size());
71 Component go10(0,1,0,0,1);
72 Component go11(1,2,0,1,2);
73 Component go12(2,3,0,2,3);
74 Component go13(3,4,0,3,4);
75 CPPUNIT_ASSERT(n->_children[0]->_comp == go10);
76 CPPUNIT_ASSERT(n->_children[1]->_comp == go11);
77 CPPUNIT_ASSERT(n->_children[2]->_comp == go12);
78 CPPUNIT_ASSERT(n->_children[3]->_comp == go13);
80 GeneOrder go2(_validPerm4.begin(),_validPerm4.end());
81 ComponentTree t2(findComponents(go2));
84 CPPUNIT_ASSERT_EQUAL((size_t)6u,count(n));
85 CPPUNIT_ASSERT_EQUAL((size_t)9u,count(n,true));
86 CPPUNIT_ASSERT_EQUAL((size_t)3u,n->_children.size());
87 Component go20(1,2,0,2,3);
88 Component go21(0,4,0,0,4);
89 Component go22(4,7,1,4,7);
90 Component go23(-15,-12,-1,8,11);
91 Component go24(-12,-9,-1,11,14);
92 Component go25(7,16,0,7,16);
93 CPPUNIT_ASSERT(n->_children[0]->_comp == go21);
94 CPPUNIT_ASSERT(n->_children[1]->_comp == go22);
95 CPPUNIT_ASSERT(n->_children[2]->_comp == go25);
96 CPPUNIT_ASSERT(n->_children[0]->_children[0]->_children[0]->_comp == go20);
97 CPPUNIT_ASSERT(n->_children[2]->_children[0]->_children[0]->_comp == go23);
98 CPPUNIT_ASSERT(n->_children[2]->_children[0]->_children[1]->_comp == go24);
102 void testMakeUnoriented (){
103 GeneOrder go(_validPerm.begin(),_validPerm.end());
104 ComponentTree t(findComponents(go));
106 ComponentTree::Node* n = t._root;
108 CPPUNIT_ASSERT_EQUAL((size_t)0u,count(n));
109 CPPUNIT_ASSERT_EQUAL((size_t)1u,count(n,true));
110 CPPUNIT_ASSERT_EQUAL((size_t)0u,n->_children.size());
112 GeneOrder go2(_validPerm4.begin(),_validPerm4.end());
113 ComponentTree t2(findComponents(go2));
117 CPPUNIT_ASSERT_EQUAL((size_t)4u,count(n));
118 CPPUNIT_ASSERT_EQUAL((size_t)6u,count(n,true));
119 CPPUNIT_ASSERT_EQUAL((size_t)2u,n->_children.size());
120 Component go22(4,7,1,4,7);
121 Component go23(-15,-12,-1,8,11);
122 Component go24(-12,-9,-1,11,14);
123 Component go25(7,16,0,7,16);
124 CPPUNIT_ASSERT(n->_children[0]->_comp == go22);
125 CPPUNIT_ASSERT(n->_children[1]->_comp == go25);
126 CPPUNIT_ASSERT(n->_children[1]->_children[0]->_children[0]->_comp == go23);
127 CPPUNIT_ASSERT(n->_children[1]->_children[0]->_children[1]->_comp == go24);
130 void testCountLeaves (){
131 GeneOrder go(_validPerm.begin(),_validPerm.end());
132 ComponentTree t(findComponents(go));
135 //CPPUNIT_ASSERT_EQUAL((size_t)0u,t.countLeaves());
137 GeneOrder go2(_validPerm4.begin(),_validPerm4.end());
138 ComponentTree t2(findComponents(go2));
140 ComponentTree::Node* n = t2._root;
141 CPPUNIT_ASSERT_EQUAL((size_t)2u,n->_children.size());
142 CPPUNIT_ASSERT_EQUAL(false,n->_children.empty());
143 CPPUNIT_ASSERT_EQUAL((size_t)3u,t2.countLeaves());
148 CPPUNIT_TEST_SUITE_REGISTRATION( TESTNAME );