test new solver functions
parent
89cd600165
commit
439b724011
|
|
@ -88,23 +88,38 @@ TEST( SymbolicSequentialSolver, inference ) {
|
||||||
// jointBayesNet
|
// jointBayesNet
|
||||||
vector<Index> js;
|
vector<Index> js;
|
||||||
js.push_back(0);
|
js.push_back(0);
|
||||||
|
js.push_back(2);
|
||||||
js.push_back(3);
|
js.push_back(3);
|
||||||
js.push_back(4);
|
|
||||||
SymbolicBayesNet::shared_ptr actualBN = solver.jointBayesNet(js);
|
SymbolicBayesNet::shared_ptr actualBN = solver.jointBayesNet(js);
|
||||||
SymbolicBayesNet expectedBN;
|
SymbolicBayesNet expectedBN;
|
||||||
expectedBN.push_front(boost::make_shared<IndexConditional>(3));
|
expectedBN.push_front(boost::make_shared<IndexConditional>(2));
|
||||||
expectedBN.push_front(boost::make_shared<IndexConditional>(4, 3));
|
expectedBN.push_front(boost::make_shared<IndexConditional>(3, 2));
|
||||||
expectedBN.push_front(boost::make_shared<IndexConditional>(0, 4));
|
expectedBN.push_front(boost::make_shared<IndexConditional>(0, 3, 2));
|
||||||
EXPECT( assert_equal(expectedBN,*actualBN));
|
EXPECT( assert_equal(expectedBN,*actualBN));
|
||||||
|
|
||||||
// jointFactorGraph
|
// jointFactorGraph
|
||||||
SymbolicFactorGraph::shared_ptr actualFG = solver.jointFactorGraph(js);
|
SymbolicFactorGraph::shared_ptr actualFG = solver.jointFactorGraph(js);
|
||||||
SymbolicFactorGraph expectedFG;
|
SymbolicFactorGraph expectedFG;
|
||||||
expectedFG.push_factor(0, 4);
|
expectedFG.push_factor(0, 3, 2);
|
||||||
expectedFG.push_factor(4, 3);
|
expectedFG.push_factor(3, 2);
|
||||||
expectedFG.push_factor(3);
|
expectedFG.push_factor(2);
|
||||||
EXPECT( assert_equal(expectedFG,(SymbolicFactorGraph)(*actualFG)));
|
EXPECT( assert_equal(expectedFG,(SymbolicFactorGraph)(*actualFG)));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
{
|
||||||
|
// conditionalBayesNet
|
||||||
|
vector<Index> js;
|
||||||
|
js.push_back(0);
|
||||||
|
js.push_back(2);
|
||||||
|
js.push_back(3);
|
||||||
|
size_t nrFrontals = 2;
|
||||||
|
SymbolicBayesNet::shared_ptr actualBN = //
|
||||||
|
solver.conditionalBayesNet(js, nrFrontals);
|
||||||
|
SymbolicBayesNet expectedBN;
|
||||||
|
expectedBN.push_front(boost::make_shared<IndexConditional>(3, 2));
|
||||||
|
expectedBN.push_front(boost::make_shared<IndexConditional>(0, 3, 2));
|
||||||
|
EXPECT( assert_equal(expectedBN,*actualBN));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/* ************************************************************************* */
|
/* ************************************************************************* */
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue