108 lines
2.9 KiB
C++
108 lines
2.9 KiB
C++
/**
|
|
* @file testIncremental.cpp
|
|
* @brief Unit tests for graph-based iSAM
|
|
* @author Michael Kaess
|
|
*/
|
|
|
|
#include <boost/assign/std/list.hpp> // for operator +=
|
|
using namespace boost::assign;
|
|
|
|
#include <CppUnitLite/TestHarness.h>
|
|
|
|
#include "SymbolicBayesNet.h"
|
|
#include "SymbolicFactorGraph.h"
|
|
#include "GaussianBayesNet.h"
|
|
#include "Ordering.h"
|
|
#include "BayesTree-inl.h"
|
|
#include "smallExample.h"
|
|
|
|
using namespace gtsam;
|
|
|
|
typedef BayesTree<SymbolicConditional> SymbolicBayesTree;
|
|
typedef BayesTree<GaussianConditional> GaussianBayesTree;
|
|
|
|
|
|
// Conditionals for ASIA example from the tutorial with A and D evidence
|
|
SymbolicConditional::shared_ptr B(new SymbolicConditional("B")), L(
|
|
new SymbolicConditional("L", "B")), E(
|
|
new SymbolicConditional("E", "L", "B")), S(new SymbolicConditional("S",
|
|
"L", "B")), T(new SymbolicConditional("T", "E", "L")), X(
|
|
new SymbolicConditional("X", "E"));
|
|
|
|
/* ************************************************************************* */
|
|
|
|
SymbolicBayesTree update(const SymbolicBayesTree& initial,
|
|
const boost::shared_ptr<SymbolicFactor>& newFactor) {
|
|
|
|
// create a factor graph with the new factor in it
|
|
SymbolicFactorGraph factorGraph;
|
|
factorGraph.push_back(newFactor);
|
|
|
|
// get the ELB clique
|
|
SymbolicBayesTree::sharedClique ELB = initial["B"];
|
|
FactorGraph<SymbolicFactor> ELB_factors(*ELB);
|
|
|
|
// add it to the factor graph
|
|
|
|
// get the SLB clique
|
|
SymbolicBayesTree::sharedClique SLB = initial["S"];
|
|
|
|
// add it to the factor graph
|
|
|
|
// create an ordering ESLB
|
|
Ordering ordering;
|
|
ordering += "S","B";
|
|
|
|
// eliminate into a Bayes net
|
|
SymbolicBayesNet bayesNet = eliminate<SymbolicFactor,SymbolicConditional>(factorGraph,ordering);
|
|
|
|
// turn back into a Bayes Tree
|
|
BayesTree<SymbolicConditional> newTree(bayesNet);
|
|
|
|
// add orphans to the bottom of the new tree
|
|
|
|
return newTree;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
TEST( BayesTree, iSAM )
|
|
{
|
|
// Create using insert
|
|
SymbolicBayesTree bayesTree;
|
|
bayesTree.insert(B);
|
|
bayesTree.insert(L);
|
|
bayesTree.insert(E);
|
|
bayesTree.insert(S);
|
|
bayesTree.insert(T);
|
|
bayesTree.insert(X);
|
|
//bayesTree.print("bayesTree");
|
|
|
|
// Create expected Bayes tree
|
|
SymbolicBayesTree expected;
|
|
expected.insert(B);
|
|
expected.insert(L);
|
|
expected.insert(S);
|
|
expected.insert(E);
|
|
expected.insert(T);
|
|
expected.insert(X);
|
|
//expected.print("expected");
|
|
|
|
// create a new factor to be inserted
|
|
list<string> keys;
|
|
keys += "B","S";
|
|
boost::shared_ptr<SymbolicFactor> newFactor(new SymbolicFactor(keys));
|
|
|
|
// do incremental inference
|
|
SymbolicBayesTree actual = update(bayesTree, newFactor);
|
|
|
|
// Check whether the same
|
|
CHECK(assert_equal(expected,actual));
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
int main() {
|
|
TestResult tr;
|
|
return TestRegistry::runAllTests(tr);
|
|
}
|
|
/* ************************************************************************* */
|