gtsam/cpp/testSymbolicBayesChain.cpp

51 lines
1.4 KiB
C++

/**
* @file testSymbolicBayesChain.cpp
* @brief Unit tests for a symbolic Bayes chain
* @author Frank Dellaert
*/
#include <boost/assign/list_inserter.hpp> // for 'insert()'
#include <boost/assign/std/list.hpp> // for operator +=
using namespace boost::assign;
#include <CppUnitLite/TestHarness.h>
#include "smallExample.h"
#include "SymbolicBayesChain.h"
#include "SymbolicFactorGraph.h"
using namespace std;
using namespace gtsam;
/* ************************************************************************* */
TEST( SymbolicBayesChain, constructor )
{
// Create manually
SymbolicConditional::shared_ptr
x2(new SymbolicConditional("l1", "x1")),
l1(new SymbolicConditional("x1")),
x1(new SymbolicConditional());
SymbolicBayesChain expected;
expected.insert("x2",x2);
expected.insert("l1",l1);
expected.insert("x1",x1);
// Create from a factor graph
LinearFactorGraph factorGraph = createLinearFactorGraph();
SymbolicFactorGraph fg(factorGraph);
// eliminate it
Ordering ordering;
ordering += "x2","l1","x1";
SymbolicBayesChain::shared_ptr actual = fg.eliminate(ordering);
CHECK(assert_equal(expected, *actual));
}
/* ************************************************************************* */
int main() {
TestResult tr;
return TestRegistry::runAllTests(tr);
}
/* ************************************************************************* */