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

Cleanup

release/4.3a0
Frank Dellaert 2022-12-28 18:10:00 -05:00
parent 1de49598af
commit fd12181ebe
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
python/gtsam/tests/test_HybridValues.py
View File

@ -1,5 +1,5 @@
""" """
GTSAM Copyright 2010-2019, Georgia Tech Research Corporation, GTSAM Copyright 2010-2022, Georgia Tech Research Corporation,
Atlanta, Georgia 30332-0415 Atlanta, Georgia 30332-0415
All Rights Reserved All Rights Reserved
@ -20,22 +20,23 @@ from gtsam.symbol_shorthand import C, X
from gtsam.utils.test_case import GtsamTestCase from gtsam.utils.test_case import GtsamTestCase
class TestHybridGaussianFactorGraph(GtsamTestCase): class TestHybridValues(GtsamTestCase):
"""Unit tests for HybridValues.""" """Unit tests for HybridValues."""
def test_basic(self): def test_basic(self):
"""Test contruction and basic methods of hybrid values.""" """Test construction and basic methods of hybrid values."""
hv1 = gtsam.HybridValues() hv1 = gtsam.HybridValues()
hv1.insert(X(0), np.ones((3,1))) hv1.insert(X(0), np.ones((3, 1)))
hv1.insert(C(0), 2) hv1.insert(C(0), 2)
hv2 = gtsam.HybridValues() hv2 = gtsam.HybridValues()
hv2.insert(C(0), 2) hv2.insert(C(0), 2)
hv2.insert(X(0), np.ones((3,1))) hv2.insert(X(0), np.ones((3, 1)))
self.assertEqual(hv1.atDiscrete(C(0)), 2) self.assertEqual(hv1.atDiscrete(C(0)), 2)
self.assertEqual(hv1.at(X(0))[0], np.ones((3,1))[0]) self.assertEqual(hv1.at(X(0))[0], np.ones((3, 1))[0])
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()