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

added test for ordering in Python

release/4.3a0
Varun Agrawal 2022-03-11 08:42:40 -05:00
parent b9f6d67fc7
commit ebc75e3a8d
1 changed files with 22 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
python/gtsam/tests/test_GaussianFactorGraph.py
View File

@ -37,9 +37,9 @@ def create_graph():
return graph, (x0, x1, x2) return graph, (x0, x1, x2)
class TestGaussianFactorGraph(GtsamTestCase): class TestGaussianFactorGraph(GtsamTestCase):
"""Tests for Gaussian Factor Graphs.""" """Tests for Gaussian Factor Graphs."""
def test_fg(self): def test_fg(self):
"""Test solving a linear factor graph""" """Test solving a linear factor graph"""
graph, X = create_graph() graph, X = create_graph()
@ -88,5 +88,22 @@ class TestGaussianFactorGraph(GtsamTestCase):
self.assertAlmostEqual(EXPECTEDM[1], m[1], delta=1e-8) self.assertAlmostEqual(EXPECTEDM[1], m[1], delta=1e-8)
self.assertAlmostEqual(EXPECTEDM[2], m[2], delta=1e-8) self.assertAlmostEqual(EXPECTEDM[2], m[2], delta=1e-8)
def test_ordering(self):
"""Test ordering"""
gfg, keys = create_graph()
ordering = gtsam.Ordering()
for key in keys[::-1]:
ordering.push_back(key)
bn = gfg.eliminateSequential(ordering)
self.assertEqual(bn.size(), 3)
keyVector = gtsam.KeyVector()
keyVector.append(keys[2])
ordering = gtsam.Ordering.ColamdConstrainedLastGaussianFactorGraph(gfg, keyVector)
bn = gfg.eliminateSequential(ordering)
self.assertEqual(bn.size(), 3)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()