]>
git.sthu.org Git - libstick.git/blob - tests/persistence.h
1 #ifndef persistence_h_Fahlaewahgaebaqu
2 #define persistence_h_Fahlaewahgaebaqu
5 #include <cpptest-suite.h>
7 #include <libstick-0.1/persistence.h>
9 using namespace libstick
;
12 class persistence_TestSuite
: public Test::Suite
{
15 typedef simplicial_complex
<3, uint32_t, double> scomplex
;
16 typedef persistence
<3, uint32_t, double> pers
;
17 typedef pers::boundary_matrix bm
;
18 typedef pers::transformation_matrix tm
;
21 scomplex ball
, ring
, torus
;
22 scomplex::simplex_order oball
, oring
, otorus
;
25 persistence_TestSuite() :
31 TEST_ADD(persistence_TestSuite::test_matrix_reduction
);
32 TEST_ADD(persistence_TestSuite::test_betti_numbers
);
36 virtual void setup() {
41 // 7 ------------------------------------- 8
43 // \ 4 ------------------------- 5 /
45 // \ \ 1 ------- 2 / /
63 scomplex::simplex ssring
[] = {
64 // dimension, faces, value...
71 /* 7 */ {1, {2, 3}, 6.01},
72 /* 8 */ {1, {3, 1}, 6.02},
73 /* 9 */ {1, {1, 2}, 6.03},
74 /* 10 */ {1, {4, 5}, 6.04},
75 /* 11 */ {1, {5, 6}, 6.05},
76 /* 12 */ {1, {6, 4}, 6.06},
77 /* 13 */ {1, {1, 4}, 6.07},
78 /* 14 */ {1, {1, 5}, 6.08},
79 /* 15 */ {2, {13, 14, 10}, 6.0801},
80 /* 16 */ {1, {3, 6}, 6.09},
81 /* 17 */ {1, {2, 5}, 6.10},
82 /* 18 */ {1, {2, 6}, 6.11},
83 /* 19 */ {2, {9, 14, 17}, 6.1101},
84 /* 20 */ {2, {7, 16, 18}, 6.1102},
85 /* 21 */ {2, {11, 17, 18}, 6.1103},
86 /* 22 */ {1, {1, 6}, 6.12},
87 /* 23 */ {2, {12, 13, 22}, 6.1201},
88 /* 24 */ {2, {8, 16, 22}, 6.1202}
90 const size_t cntssring
= sizeof(ssring
)/sizeof(scomplex::simplex
);
91 ring
.add_simplices(ssring
, cntssring
);
94 scomplex::simplex sstorus
[] = {
95 // dimension, faces, value...
99 /* 28 */ {1, {25, 26}, 9.01},
100 /* 29 */ {1, {26, 27}, 9.02},
101 /* 30 */ {1, {25, 27}, 9.03},
102 /* 31 */ {1, {25, 1}, 9.04},
103 /* 32 */ {1, {26, 1}, 9.05},
104 /* 33 */ {2, {28, 31, 32}, 9.0501},
105 /* 34 */ {1, {27, 1}, 9.06},
106 /* 35 */ {2, {30, 31, 34}, 9.0601},
107 /* 36 */ {1, {27, 3}, 9.07},
108 /* 37 */ {2, {36, 34, 8}, 9.0701},
109 /* 38 */ {1, {27, 2}, 9.08},
110 /* 39 */ {1, {26, 2}, 9.09},
111 /* 40 */ {2, {9, 32, 39}, 9.0901},
112 /* 41 */ {2, {29, 38, 39}, 9.0902},
113 /* 42 */ {2, {7, 38, 36}, 9.0903},
114 /* 43 */ {1, {4, 25}, 9.10},
115 /* 44 */ {1, {5, 26}, 9.11},
116 /* 45 */ {1, {6, 27}, 9.12},
117 /* 46 */ {1, {4, 26}, 9.13},
118 /* 47 */ {1, {4, 27}, 9.14},
119 /* 48 */ {1, {5, 27}, 9.15},
120 /* 49 */ {2, {43, 47, 30}, 9.1501},
121 /* 50 */ {2, {12, 45, 47}, 9.1502},
122 /* 51 */ {2, {29, 44, 48}, 9.1503},
123 /* 52 */ {2, {48, 11, 45}, 9.1504},
124 /* 53 */ {2, {10, 46, 44}, 9.1505},
125 /* 54 */ {2, {43, 46, 28}, 9.1506},
127 const size_t cntsstorus
= sizeof(sstorus
)/sizeof(scomplex::simplex
);
129 torus
.add_simplices(sstorus
, cntsstorus
);
132 scomplex::simplex ssball
[] = {
133 // dimension, faces, value...
147 {2, {6, 10, 11}, 14},
148 {2, {7, 11, 12}, 15},
149 {2, {8, 12, 13}, 16},
150 {2, {9, 13, 10}, 17},
154 {2, {18, 10, 12}, 21},
155 {3, {21, 14, 15, 19}, 22},
156 {3, {21, 16, 17, 20}, 23},
158 const size_t cntssball
= sizeof(ssball
)/sizeof(scomplex::simplex
);
159 ball
.add_simplices(ssball
, cntssball
);
163 virtual void tear_down() {
166 void test_matrix_reduction() {
168 ringp
.compute_matrices();
169 const bm
&ringbm
= ringp
.get_boundary_matrix();
170 const bm
&ringrbm
= ringp
.get_reduced_boundary_matrix();
171 const tm
&ringtm
= ringp
.get_transformation_matrix();
172 TEST_ASSERT(ringrbm
== ringbm
* ringtm
);
175 torusp
.compute_matrices();
176 const bm
&torusbm
= torusp
.get_boundary_matrix();
177 const bm
&torusrbm
= torusp
.get_reduced_boundary_matrix();
178 const tm
&torustm
= torusp
.get_transformation_matrix();
179 TEST_ASSERT(torusrbm
== torusbm
* torustm
);
181 uint32_t torusrbme_coords
[][2] = {
182 {0, 1}, {2, 7}, {3, 7}, {1, 8}, {2, 8}, {4, 10}, {5, 10}, {4, 11}, {6, 11}, {1, 13},
183 {4, 13}, {10, 15}, {13, 15}, {14, 15}, {9, 19}, {10, 19}, {13, 19}, {17, 19}, {7, 20},
184 {16, 20}, {18, 20}, {7, 21}, {9, 21}, {10, 21}, {11, 21}, {13, 21}, {16, 21}, {12, 23},
185 {13, 23}, {22, 23}, {7, 24}, {8, 24}, {9, 24}, {10, 24}, {11, 24}, {12, 24}, {25, 28},
186 {26, 28}, {25, 29}, {27, 29}, {1, 31}, {25, 31}, {28, 33}, {31, 33}, {32, 33}, {30, 35},
187 {31, 35}, {34, 35}, {8, 37}, {30, 37}, {31, 37}, {36, 37}, {9, 40}, {28, 40}, {31, 40},
188 {39, 40}, {9, 41}, {28, 41}, {29, 41}, {31, 41}, {38, 41}, {7, 42}, {8, 42}, {9, 42},
189 {28, 42}, {29, 42}, {30, 42}, {7, 49}, {8, 49}, {9, 49}, {28, 49}, {29, 49}, {43, 49},
190 {47, 49}, {10, 50}, {11, 50}, {28, 50}, {29, 50}, {43, 50}, {45, 50}, {29, 51}, {44, 51},
191 {48, 51}, {10, 52}, {28, 52}, {43, 52}, {44, 52}, {28, 53}, {43, 53}, {46, 53 }
193 bm
torusrbme(torusbm
.size());
194 torusrbme
.set_all(torusrbme_coords
, sizeof(torusrbme_coords
)/(2*sizeof(uint32_t)), true);
195 TEST_ASSERT(torusrbme
== torusrbm
);
199 ballp
.compute_matrices();
200 const bm
&ballbm
= ballp
.get_boundary_matrix();
201 const bm
&ballrbm
= ballp
.get_reduced_boundary_matrix();
202 const tm
&balltm
= ballp
.get_transformation_matrix();
203 TEST_ASSERT(ballrbm
== ballbm
* balltm
);
205 uint32_t ballrbme_coords
[][2] = {
206 {0, 1}, {1, 6}, {2, 6}, {1, 7}, {3, 7}, {1, 8}, {4, 8}, {1, 10}, {5, 10}, {6, 14},
207 {10, 14}, {11, 14}, {6, 15}, {7, 15}, {10, 15}, {12, 15}, {6, 16}, {7, 16}, {8, 16},
208 {10, 16}, {13, 16}, {6, 17}, {7, 17}, {8, 17}, {9, 17}, {6, 19}, {7, 19}, {18, 19},
209 {14, 22}, {15, 22}, {19, 22}, {21, 22}, {14, 23}, {15, 23}, {16, 23}, {17, 23}, {19, 23},
212 bm
ballrbme(ballrbm
.size());
213 ballrbme
.set_all(ballrbme_coords
, sizeof(ballrbme_coords
)/(2*sizeof(uint32_t)), true);
214 TEST_ASSERT(ballrbme
== ballrbm
);
216 //std::cout << std::endl;
217 //ballp.interpret_reduction(std::cout);
218 //torusp.compute_diagrams();
219 //torusp.interpret_persistence(std::cout);
220 //std::cout << std::endl;
221 //std::cout << std::endl;
224 void test_betti_numbers() {
226 ringp
.compute_diagrams();
228 TEST_ASSERT(ringp
.get_persistence_diagram(-1).betti() == 0);
229 TEST_ASSERT(ringp
.get_persistence_diagram(0).betti() == 0);
230 TEST_ASSERT(ringp
.get_persistence_diagram(1).betti() == 1);
231 TEST_ASSERT(ringp
.get_persistence_diagram(2).betti() == 0);
232 TEST_ASSERT(ringp
.get_persistence_diagram(3).betti() == 0);
233 for (unsigned d
=0; d
< 3; ++d
)
234 TEST_ASSERT(ringp
.get_persistence_diagram(d
).persistent_betti(oring
.size()-1, oring
.size()-1) ==
235 ringp
.get_persistence_diagram(d
).betti());
239 torusp
.compute_diagrams();
241 TEST_ASSERT(torusp
.get_persistence_diagram(-1).betti() == 0);
242 TEST_ASSERT(torusp
.get_persistence_diagram(0).betti() == 0);
243 TEST_ASSERT(torusp
.get_persistence_diagram(1).betti() == 2);
244 TEST_ASSERT(torusp
.get_persistence_diagram(2).betti() == 1);
245 TEST_ASSERT(torusp
.get_persistence_diagram(3).betti() == 0);
246 for (unsigned d
=0; d
< 3; ++d
)
247 TEST_ASSERT(torusp
.get_persistence_diagram(d
).persistent_betti(otorus
.size()-1, otorus
.size()-1) ==
248 torusp
.get_persistence_diagram(d
).betti());
250 // Torus was made from ring. Hence, persistent betti numbers must correspond to betti numbers of ring.
251 for (unsigned d
=0; d
< 3; ++d
)
252 TEST_ASSERT(torusp
.get_persistence_diagram(d
).persistent_betti(oring
.size()-1, oring
.size()-1) ==
253 ringp
.get_persistence_diagram(d
).betti());
257 ballp
.compute_diagrams();
259 //std::cout << std::endl;
260 //ballp.interpret_reduction(std::cout);
261 //std::cout << std::endl;
263 TEST_ASSERT(ballp
.get_persistence_diagram(-1).betti() == 0);
264 TEST_ASSERT(ballp
.get_persistence_diagram(0).betti() == 0);
265 TEST_ASSERT(ballp
.get_persistence_diagram(1).betti() == 0);
266 TEST_ASSERT(ballp
.get_persistence_diagram(2).betti() == 0);
267 TEST_ASSERT(ballp
.get_persistence_diagram(3).betti() == 0);
268 for (unsigned d
=0; d
< 3; ++d
)
269 TEST_ASSERT(ballp
.get_persistence_diagram(d
).persistent_betti(oball
.size()-1, oball
.size()-1) ==
270 ballp
.get_persistence_diagram(d
).betti());
272 // Before we inserted the two tetrahedra and the interior triangle, we had a sphere.
273 TEST_ASSERT(ballp
.get_persistence_diagram(-1).persistent_betti(oball
.size()-4, oball
.size()-4) == 0);
274 TEST_ASSERT(ballp
.get_persistence_diagram(0).persistent_betti(oball
.size()-4, oball
.size()-4) == 0);
275 TEST_ASSERT(ballp
.get_persistence_diagram(1).persistent_betti(oball
.size()-4, oball
.size()-4) == 0);
276 TEST_ASSERT(ballp
.get_persistence_diagram(2).persistent_betti(oball
.size()-4, oball
.size()-4) == 1);
277 TEST_ASSERT(ballp
.get_persistence_diagram(3).persistent_betti(oball
.size()-4, oball
.size()-4) == 0);
279 // Before we inserted the two tetrahedra, we had a double-sphere.
280 TEST_ASSERT(ballp
.get_persistence_diagram(-1).persistent_betti(oball
.size()-3, oball
.size()-3) == 0);
281 TEST_ASSERT(ballp
.get_persistence_diagram(0).persistent_betti(oball
.size()-3, oball
.size()-3) == 0);
282 TEST_ASSERT(ballp
.get_persistence_diagram(1).persistent_betti(oball
.size()-3, oball
.size()-3) == 0);
283 TEST_ASSERT(ballp
.get_persistence_diagram(2).persistent_betti(oball
.size()-3, oball
.size()-3) == 2);
284 TEST_ASSERT(ballp
.get_persistence_diagram(3).persistent_betti(oball
.size()-3, oball
.size()-3) == 0);