fix testHybridGaussianFactorGraph

release/4.3a0
Varun Agrawal 2025-01-01 22:10:59 -05:00
parent fcc56f5de6
commit f80a3a1a1d
1 changed files with 5 additions and 3 deletions

View File

@ -650,7 +650,7 @@ TEST(HybridGaussianFactorGraph, EliminateTiny1) {
mode, std::vector{conditional0, conditional1}); mode, std::vector{conditional0, conditional1});
// Add prior on mode. // Add prior on mode.
expectedBayesNet.emplace_shared<DiscreteConditional>(mode, "74/26"); expectedBayesNet.emplace_shared<DiscreteTableConditional>(mode, "74/26");
// Test elimination // Test elimination
const auto posterior = fg.eliminateSequential(); const auto posterior = fg.eliminateSequential();
@ -700,7 +700,7 @@ TEST(HybridGaussianFactorGraph, EliminateTiny1Swapped) {
m1, std::vector{conditional0, conditional1}); m1, std::vector{conditional0, conditional1});
// Add prior on m1. // Add prior on m1.
expectedBayesNet.emplace_shared<DiscreteConditional>( expectedBayesNet.emplace_shared<DiscreteTableConditional>(
m1, "0.188638/0.811362"); m1, "0.188638/0.811362");
// Test elimination // Test elimination
@ -737,7 +737,9 @@ TEST(HybridGaussianFactorGraph, EliminateTiny2) {
mode, std::vector{conditional0, conditional1}); mode, std::vector{conditional0, conditional1});
// Add prior on mode. // Add prior on mode.
expectedBayesNet.emplace_shared<DiscreteConditional>(mode, "23/77"); // Since this is the only discrete conditional, it is added as a
// DiscreteTableConditional.
expectedBayesNet.emplace_shared<DiscreteTableConditional>(mode, "23/77");
// Test elimination // Test elimination
const auto posterior = fg.eliminateSequential(); const auto posterior = fg.eliminateSequential();