diff --git a/python/gtsam/tests/test_HybridNonlinearFactorGraph.py b/python/gtsam/tests/test_HybridNonlinearFactorGraph.py index 3ac0d5c6f..739abcc09 100644 --- a/python/gtsam/tests/test_HybridNonlinearFactorGraph.py +++ b/python/gtsam/tests/test_HybridNonlinearFactorGraph.py @@ -18,6 +18,7 @@ import gtsam import numpy as np from gtsam.symbol_shorthand import C, X from gtsam.utils.test_case import GtsamTestCase +from gtsam import BetweenFactorPoint3, noiseModel, PriorFactorPoint3, Point3 class TestHybridGaussianFactorGraph(GtsamTestCase): @@ -27,20 +28,22 @@ class TestHybridGaussianFactorGraph(GtsamTestCase): nlfg = gtsam.HybridNonlinearFactorGraph() dk = gtsam.DiscreteKeys() dk.push_back((10, 2)) - nlfg.add(gtsam.BetweenFactorPoint3(1, 2, gtsam.Point3(1, 2, 3), gtsam.noiseModel.Diagonal.Variances([1, 1, 1]))) - nlfg.add( - gtsam.PriorFactorPoint3(2, gtsam.Point3(1, 2, 3), gtsam.noiseModel.Diagonal.Variances([0.5, 0.5, 0.5]))) + nlfg.push_back(BetweenFactorPoint3(1, 2, Point3( + 1, 2, 3), noiseModel.Diagonal.Variances([1, 1, 1]))) + nlfg.push_back( + PriorFactorPoint3(2, Point3(1, 2, 3), + noiseModel.Diagonal.Variances([0.5, 0.5, 0.5]))) nlfg.push_back( gtsam.MixtureFactor([1], dk, [ - gtsam.PriorFactorPoint3(1, gtsam.Point3(0, 0, 0), - gtsam.noiseModel.Unit.Create(3)), - gtsam.PriorFactorPoint3(1, gtsam.Point3(1, 2, 1), - gtsam.noiseModel.Unit.Create(3)) + PriorFactorPoint3(1, Point3(0, 0, 0), + noiseModel.Unit.Create(3)), + PriorFactorPoint3(1, Point3(1, 2, 1), + noiseModel.Unit.Create(3)) ])) - nlfg.add(gtsam.DecisionTreeFactor((10, 2), "1 3")) + nlfg.push_back(gtsam.DecisionTreeFactor((10, 2), "1 3")) values = gtsam.Values() - values.insert_point3(1, gtsam.Point3(0, 0, 0)) - values.insert_point3(2, gtsam.Point3(2, 3, 1)) + values.insert_point3(1, Point3(0, 0, 0)) + values.insert_point3(2, Point3(2, 3, 1)) hfg = nlfg.linearize(values) o = gtsam.Ordering() o.push_back(1)