diff --git a/gtsam/discrete/tests/testDecisionTree.cpp b/gtsam/discrete/tests/testDecisionTree.cpp index fb49908e2..71daf261d 100644 --- a/gtsam/discrete/tests/testDecisionTree.cpp +++ b/gtsam/discrete/tests/testDecisionTree.cpp @@ -524,8 +524,6 @@ TEST(DecisionTree, ApplyWithAssignment) { DT prunedTree = tree.apply(pruner); DT expectedTree(keys, "0 0 0 0 5 6 7 8"); - // expectedTree.print(); - // prunedTree.print(); EXPECT(assert_equal(expectedTree, prunedTree)); size_t count = 0; @@ -539,24 +537,6 @@ TEST(DecisionTree, ApplyWithAssignment) { EXPECT_LONGS_EQUAL(5, count); } -/* ************************************************************************** */ -// Test number of assignments. -TEST(DecisionTree, Constructor) { - using gtsam::symbol_shorthand::M; - - std::vector probs = {0, 0, 1, 2}; - - DiscreteKeys keys{{M(1), 2}, {M(0), 2}}; - DecisionTree dt1(keys, probs); - EXPECT_LONGS_EQUAL(4, dt1.nrAssignments()); - dt1.print("", DefaultKeyFormatter, [](double x) { return std::to_string(x);}); - - DiscreteKeys keys2{{M(0), 2}, {M(1), 2}}; - DecisionTree dt2(keys2, probs); - std::cout << "\n" << std::endl; - dt2.print("", DefaultKeyFormatter, [](double x) { return std::to_string(x);}); -} - /* ************************************************************************** */ // Test number of assignments. TEST(DecisionTree, NrAssignments2) {