From 31bbf513f46cbf0909f5991dbbe5a246e195f6d5 Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Sat, 27 Aug 2022 19:23:27 -0400
Subject: [PATCH] add discrete factors to HybridBayesTree test

---
 gtsam/hybrid/tests/testHybridBayesTree.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/gtsam/hybrid/tests/testHybridBayesTree.cpp b/gtsam/hybrid/tests/testHybridBayesTree.cpp
index 3584d7db9..d457e6b74 100644
--- a/gtsam/hybrid/tests/testHybridBayesTree.cpp
+++ b/gtsam/hybrid/tests/testHybridBayesTree.cpp
@@ -51,6 +51,11 @@ TEST(HybridBayesTree, OptimizeAssignment) {
     graph1.push_back(s.linearizedFactorGraph.at(i));
   }
 
+  // Add the discrete factors
+  for (size_t i = 7; i <= 9; i++) {
+    graph1.push_back(s.linearizedFactorGraph.at(i));
+  }
+
   isam.update(graph1);
 
   DiscreteValues assignment;