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

move create to the top

release/4.3a0
Varun Agrawal 2025-01-05 08:45:11 -05:00
parent 6b6283c151
commit a142556c52
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
gtsam/discrete/tests/testDecisionTreeFactor.cpp
View File

@ -30,6 +30,12 @@
using namespace std; using namespace std;
using namespace gtsam; using namespace gtsam;
/** Convert Signature into CPT */
DecisionTreeFactor create(const Signature& signature) {
DecisionTreeFactor p(signature.discreteKeys(), signature.cpt());
return p;
}
/* ************************************************************************* */ /* ************************************************************************* */
TEST(DecisionTreeFactor, ConstructorsMatch) { TEST(DecisionTreeFactor, ConstructorsMatch) {
// Declare two keys // Declare two keys
@ -69,15 +75,6 @@ TEST(DecisionTreeFactor, constructors) {
EXPECT_DOUBLES_EQUAL(0.8, f4(x121), 1e-9); EXPECT_DOUBLES_EQUAL(0.8, f4(x121), 1e-9);
} }
/* ************************************************************************* */
TEST(DecisionTreeFactor, Divide) {
DiscreteKey A(0, 2), S(1, 2);
DecisionTreeFactor pA = create(A % "99/1"), pS = create(S % "50/50");
DecisionTreeFactor joint = pA * pS;
DecisionTreeFactor s = joint / pA;
EXPECT(assert_equal(pS, s));
}
/* ************************************************************************* */ /* ************************************************************************* */
TEST(DecisionTreeFactor, Error) { TEST(DecisionTreeFactor, Error) {
// Declare a bunch of keys // Declare a bunch of keys
@ -114,6 +111,15 @@ TEST(DecisionTreeFactor, multiplication) {
CHECK(assert_equal(expected2, actual)); CHECK(assert_equal(expected2, actual));
} }
/* ************************************************************************* */
TEST(DecisionTreeFactor, Divide) {
DiscreteKey A(0, 2), S(1, 2);
DecisionTreeFactor pA = create(A % "99/1"), pS = create(S % "50/50");
DecisionTreeFactor joint = pA * pS;
DecisionTreeFactor s = joint / pA;
EXPECT(assert_equal(pS, s));
}
/* ************************************************************************* */ /* ************************************************************************* */
TEST(DecisionTreeFactor, sum_max) { TEST(DecisionTreeFactor, sum_max) {
DiscreteKey v0(0, 3), v1(1, 2); DiscreteKey v0(0, 3), v1(1, 2);
@ -226,12 +232,6 @@ void maybeSaveDotFile(const DecisionTreeFactor& f, const string& filename) {
#endif #endif
} }
/** Convert Signature into CPT */
DecisionTreeFactor create(const Signature& signature) {
DecisionTreeFactor p(signature.discreteKeys(), signature.cpt());
return p;
}
/* ************************************************************************* */ /* ************************************************************************* */
// test Asia Joint // test Asia Joint
TEST(DecisionTreeFactor, joint) { TEST(DecisionTreeFactor, joint) {