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

update test

release/4.3a0
Varun Agrawal 2025-01-04 14:39:53 -05:00
parent d3780158b1
commit 14f32544d2
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
gtsam/hybrid/tests/testHybridBayesNet.cpp
View File

@ -451,8 +451,7 @@ TEST(HybridBayesNet, UpdateDiscreteConditionals) {
DiscreteConditional joint; DiscreteConditional joint;
for (auto&& conditional : posterior->discreteMarginal()) { for (auto&& conditional : posterior->discreteMarginal()) {
// The last discrete conditional may be a TableDistribution // The last discrete conditional may be a TableDistribution
if (auto dtc = if (auto dtc = std::dynamic_pointer_cast<TableDistribution>(conditional)) {
std::dynamic_pointer_cast<TableDistribution>(conditional)) {
DiscreteConditional dc(dtc->nrFrontals(), dtc->toDecisionTreeFactor()); DiscreteConditional dc(dtc->nrFrontals(), dtc->toDecisionTreeFactor());
joint = joint * dc; joint = joint * dc;
} else { } else {
@ -461,7 +460,8 @@ TEST(HybridBayesNet, UpdateDiscreteConditionals) {
} }
size_t maxNrLeaves = 3; size_t maxNrLeaves = 3;
auto prunedDecisionTree = *joint.prune(maxNrLeaves); DiscreteConditional prunedDecisionTree(joint);
prunedDecisionTree.prune(maxNrLeaves);
#ifdef GTSAM_DT_MERGING #ifdef GTSAM_DT_MERGING
EXPECT_LONGS_EQUAL(maxNrLeaves + 2 /*2 zero leaves*/, EXPECT_LONGS_EQUAL(maxNrLeaves + 2 /*2 zero leaves*/,