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

fix python test

release/4.3a0
Varun Agrawal 2025-01-04 16:58:18 -05:00
parent 9a356f102e
commit aba691d3d6
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
python/gtsam/tests/test_HybridFactorGraph.py
View File

@ -13,14 +13,14 @@ Author: Fan Jiang, Varun Agrawal, Frank Dellaert
import unittest import unittest
import numpy as np import numpy as np
from gtsam.symbol_shorthand import C, M, X, Z
from gtsam.utils.test_case import GtsamTestCase
import gtsam import gtsam
from gtsam import (DiscreteConditional, GaussianConditional, from gtsam import (DiscreteConditional, GaussianConditional, HybridBayesNet,
HybridBayesNet, HybridGaussianConditional, HybridGaussianConditional, HybridGaussianFactor,
HybridGaussianFactor, HybridGaussianFactorGraph, HybridGaussianFactorGraph, HybridValues, JacobianFactor,
HybridValues, JacobianFactor, noiseModel) TableDistribution, noiseModel)
from gtsam.symbol_shorthand import C, M, X, Z
from gtsam.utils.test_case import GtsamTestCase
DEBUG_MARGINALS = False DEBUG_MARGINALS = False
@ -51,7 +51,7 @@ class TestHybridGaussianFactorGraph(GtsamTestCase):
self.assertEqual(len(hybridCond.keys()), 2) self.assertEqual(len(hybridCond.keys()), 2)
discrete_conditional = hbn.at(hbn.size() - 1).inner() discrete_conditional = hbn.at(hbn.size() - 1).inner()
self.assertIsInstance(discrete_conditional, DiscreteConditional) self.assertIsInstance(discrete_conditional, TableDistribution)
def test_optimize(self): def test_optimize(self):
"""Test construction of hybrid factor graph.""" """Test construction of hybrid factor graph."""