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

Removed copy/paste in unit tests

release/4.3a0
Frank Dellaert 2009-11-22 18:12:29 +00:00
parent d4aaa5d114
commit 9da2f16f4d
1 changed files with 23 additions and 63 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

86
cpp/testBayesTree.cpp
View File

@ -22,11 +22,25 @@ typedef BayesTree<SymbolicConditional> SymbolicBayesTree;
typedef BayesTree<GaussianConditional> GaussianBayesTree; typedef BayesTree<GaussianConditional> GaussianBayesTree;
// Conditionals for ASIA example from the tutorial with A and D evidence // Conditionals for ASIA example from the tutorial with A and D evidence
SymbolicConditional::shared_ptr B(new SymbolicConditional("B")), L( SymbolicConditional::shared_ptr
new SymbolicConditional("L", "B")), E( B(new SymbolicConditional("B")),
new SymbolicConditional("E", "L", "B")), S(new SymbolicConditional("S", L(new SymbolicConditional("L", "B")),
"L", "B")), T(new SymbolicConditional("T", "E", "L")), X( E(new SymbolicConditional("E", "B", "L")),
new SymbolicConditional("X", "E")); S(new SymbolicConditional("S", "L", "B")),
T(new SymbolicConditional("T", "E", "L")),
X(new SymbolicConditional("X", "E"));
// Bayes Tree for Asia example
SymbolicBayesTree createAsiaSymbolicBayesTree() {
SymbolicBayesTree bayesTree;
bayesTree.insert(B);
bayesTree.insert(L);
bayesTree.insert(E);
bayesTree.insert(S);
bayesTree.insert(T);
bayesTree.insert(X);
return bayesTree;
}
/* ************************************************************************* */ /* ************************************************************************* */
TEST( BayesTree, Front ) TEST( BayesTree, Front )
@ -45,13 +59,7 @@ TEST( BayesTree, Front )
TEST( BayesTree, constructor ) TEST( BayesTree, constructor )
{ {
// Create using insert // Create using insert
SymbolicBayesTree bayesTree; SymbolicBayesTree bayesTree = createAsiaSymbolicBayesTree();
bayesTree.insert(B);
bayesTree.insert(L);
bayesTree.insert(E);
bayesTree.insert(S);
bayesTree.insert(T);
bayesTree.insert(X);
// Check Size // Check Size
LONGS_EQUAL(4,bayesTree.size()); LONGS_EQUAL(4,bayesTree.size());
@ -354,23 +362,7 @@ TEST( BayesTree, removePath )
/* ************************************************************************* */ /* ************************************************************************* */
TEST( BayesTree, removePath2 ) TEST( BayesTree, removePath2 )
{ {
// Conditionals for ASIA example from the tutorial with A and D evidence SymbolicBayesTree bayesTree = createAsiaSymbolicBayesTree();
SymbolicConditional::shared_ptr
B(new SymbolicConditional("B")),
L(new SymbolicConditional("L", "B")),
E(new SymbolicConditional("E", "B", "L")),
S(new SymbolicConditional("S", "L", "B")),
T(new SymbolicConditional("T", "E", "L")),
X(new SymbolicConditional("X", "E"));
// Create using insert
SymbolicBayesTree bayesTree;
bayesTree.insert(B);
bayesTree.insert(L);
bayesTree.insert(E);
bayesTree.insert(S);
bayesTree.insert(T);
bayesTree.insert(X);
// Call remove-path with clique S // Call remove-path with clique S
FactorGraph<SymbolicFactor> factors; FactorGraph<SymbolicFactor> factors;
@ -391,23 +383,7 @@ TEST( BayesTree, removePath2 )
/* ************************************************************************* */ /* ************************************************************************* */
TEST( BayesTree, removePath3 ) TEST( BayesTree, removePath3 )
{ {
// Conditionals for ASIA example from the tutorial with A and D evidence SymbolicBayesTree bayesTree = createAsiaSymbolicBayesTree();
SymbolicConditional::shared_ptr
B(new SymbolicConditional("B")),
L(new SymbolicConditional("L", "B")),
E(new SymbolicConditional("E", "B", "L")),
S(new SymbolicConditional("S", "L", "B")),
T(new SymbolicConditional("T", "E", "L")),
X(new SymbolicConditional("X", "E"));
// Create using insert
SymbolicBayesTree bayesTree;
bayesTree.insert(B);
bayesTree.insert(L);
bayesTree.insert(E);
bayesTree.insert(S);
bayesTree.insert(T);
bayesTree.insert(X);
// Call remove-path with clique S // Call remove-path with clique S
FactorGraph<SymbolicFactor> factors; FactorGraph<SymbolicFactor> factors;
@ -429,23 +405,7 @@ TEST( BayesTree, removePath3 )
/* ************************************************************************* */ /* ************************************************************************* */
TEST( BayesTree, removeTop ) TEST( BayesTree, removeTop )
{ {
// Conditionals for ASIA example from the tutorial with A and D evidence SymbolicBayesTree bayesTree = createAsiaSymbolicBayesTree();
SymbolicConditional::shared_ptr
B(new SymbolicConditional("B")),
L(new SymbolicConditional("L", "B")),
E(new SymbolicConditional("E", "B", "L")),
S(new SymbolicConditional("S", "L", "B")),
T(new SymbolicConditional("T", "E", "L")),
X(new SymbolicConditional("X", "E"));
// Create using insert
SymbolicBayesTree bayesTree;
bayesTree.insert(B);
bayesTree.insert(L);
bayesTree.insert(E);
bayesTree.insert(S);
bayesTree.insert(T);
bayesTree.insert(X);
// create a new factor to be inserted // create a new factor to be inserted
list<string> keys; list<string> keys;