From f4420f2c8dbed17efe506105485a1a5572a67290 Mon Sep 17 00:00:00 2001 From: Varun Agrawal Date: Thu, 29 Dec 2022 08:40:46 +0530 Subject: [PATCH] add mixture to bayesnet and fix double assert bug --- python/gtsam/tests/test_HybridBayesNet.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/python/gtsam/tests/test_HybridBayesNet.py b/python/gtsam/tests/test_HybridBayesNet.py index 13ac3a3e2..66cddf05e 100644 --- a/python/gtsam/tests/test_HybridBayesNet.py +++ b/python/gtsam/tests/test_HybridBayesNet.py @@ -48,7 +48,7 @@ class TestHybridBayesNet(GtsamTestCase): # Create hybrid Bayes net. bayesNet = HybridBayesNet() bayesNet.addGaussian(gc) - # bayesNet.addMixture(gm) + bayesNet.addMixture(gm) bayesNet.addDiscrete(Asia, "99/1") # Create values at which to evaluate. @@ -59,10 +59,10 @@ class TestHybridBayesNet(GtsamTestCase): conditionalProbability = gc.evaluate(values.continuous()) mixtureProbability = conditional0.evaluate(values.continuous()) - assert self.assertAlmostEqual(conditionalProbability * - mixtureProbability * 0.99, - bayesNet.evaluate(values), - places=5) + self.assertAlmostEqual(conditionalProbability * mixtureProbability * + 0.99, + bayesNet.evaluate(values), + places=5) if __name__ == "__main__":