12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Split test

release/4.3a0
Frank Dellaert 2024-09-23 17:53:43 -07:00
parent 4c7d3b5a50
commit 502101100e
1 changed files with 22 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
gtsam/discrete/tests/testAlgebraicDecisionTree.cpp
View File

@ -134,8 +134,7 @@ ADT create(const Signature& signature) {
}
/* ************************************************************************* */
// test Asia Joint
TEST(ADT, joint) {
namespace asiaCPTs {
DiscreteKey A(0, 2), S(1, 2), T(2, 2), L(3, 2), B(4, 2), E(5, 2), X(6, 2),
D(7, 2);
@ -147,6 +146,12 @@ TEST(ADT, joint) {
ADT pE = create((E | T, L) = "F T T T");
ADT pX = create(X | E = "95/5 2/98");
ADT pD = create((D | E, B) = "9/1 2/8 3/7 1/9");
} // namespace asiaCPTs
/* ************************************************************************* */
// test Asia Joint
TEST(ADT, joint) {
using namespace asiaCPTs;
// Create joint
resetCounts();
@ -172,6 +177,11 @@ TEST(ADT, joint) {
EXPECT_LONGS_EQUAL(508, muls);
#endif
printCounts("Asia joint");
}
/* ************************************************************************* */
TEST(ADT, combine) {
using namespace asiaCPTs;
// Form P(A,S,T,L) = P(A) P(S) P(T|A) P(L|S)
ADT pASTL = pA;
@ -187,7 +197,7 @@ TEST(ADT, joint) {
}
/* ************************************************************************* */
// test Inference with joint
// test Inference with joint, created using different ordering
TEST(ADT, inference) {
DiscreteKey A(0, 2), D(1, 2), //
B(2, 2), L(3, 2), E(4, 2), S(5, 2), T(6, 2), X(7, 2);
@ -248,7 +258,6 @@ TEST(ADT, inference) {
TEST(ADT, factor_graph) {
DiscreteKey B(0, 2), L(1, 2), E(2, 2), S(3, 2), T(4, 2), X(5, 2);
resetCounts();
ADT pS = create(S % "50/50");
ADT pT = create(T % "95/5");
ADT pL = create(L | S = "99/1 90/10");
@ -256,7 +265,6 @@ TEST(ADT, factor_graph) {
ADT pX = create(X | E = "95/5 2/98");
ADT pD = create(B | E = "1/8 7/9");
ADT pB = create(B | S = "70/30 40/60");
printCounts("Create CPTs");
// Create joint
resetCounts();