diff --git a/gtsam/discrete/tests/testSerializationDiscrete.cpp b/gtsam/discrete/tests/testSerializationDiscrete.cpp index df7df0b7e..9d15d0536 100644 --- a/gtsam/discrete/tests/testSerializationDiscrete.cpp +++ b/gtsam/discrete/tests/testSerializationDiscrete.cpp @@ -20,6 +20,7 @@ #include #include #include +#include #include using namespace std; @@ -32,6 +33,7 @@ BOOST_CLASS_EXPORT_GUID(Tree::Leaf, "gtsam_DecisionTreeStringInt_Leaf") BOOST_CLASS_EXPORT_GUID(Tree::Choice, "gtsam_DecisionTreeStringInt_Choice") BOOST_CLASS_EXPORT_GUID(DecisionTreeFactor, "gtsam_DecisionTreeFactor"); +BOOST_CLASS_EXPORT_GUID(TableFactor, "gtsam_TableFactor"); using ADT = AlgebraicDecisionTree; BOOST_CLASS_EXPORT_GUID(ADT, "gtsam_AlgebraicDecisionTree"); @@ -79,6 +81,19 @@ TEST(DiscreteSerialization, DecisionTreeFactor) { EXPECT(equalsBinary(f)); } +/* ************************************************************************* */ +// Check serialization for TableFactor +TEST(DiscreteSerialization, TableFactor) { + using namespace serializationTestHelpers; + + DiscreteKey A(Symbol('x', 1), 3); + TableFactor tf(A % "1/2/2"); + + EXPECT(equalsObj(tf)); + EXPECT(equalsXML(tf)); + EXPECT(equalsBinary(tf)); +} + /* ************************************************************************* */ // Check serialization for DiscreteConditional & DiscreteDistribution TEST(DiscreteSerialization, DiscreteConditional) {