From cc7ed2d152655365a24a443454800ac5a74ab124 Mon Sep 17 00:00:00 2001
From: Ankur Roy Chowdhury <ankur.roy@verdantrobotics.com>
Date: Mon, 23 Jan 2023 16:30:39 -0800
Subject: [PATCH] add unit test for 'assignWeights'

---
 tests/testSubgraphSolver.cpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tests/testSubgraphSolver.cpp b/tests/testSubgraphSolver.cpp
index 245572896..9a6f2c76b 100644
--- a/tests/testSubgraphSolver.cpp
+++ b/tests/testSubgraphSolver.cpp
@@ -129,6 +129,19 @@ TEST( SubgraphSolver, constructor3 )
   DOUBLES_EQUAL(0.0, error(Ab, optimized), 1e-5);
 }
 
+/* ************************************************************************* */
+TEST(SubgraphBuilder, utilsAssignWeights)
+{
+  const auto [g, _] = example::planarGraph(N); // A*x-b
+  const auto weights = utils::assignWeights(g, gtsam::SubgraphBuilderParameters::SkeletonWeight::EQUAL);
+
+  EXPECT(weights.size() == g.size());
+  for (const auto &i : weights)
+  {
+    EXPECT_DOUBLES_EQUAL(weights[i], 1.0, 1e-12);
+  }
+}
+
 /* ************************************************************************* */
 TEST(SubgraphBuilder, utilsKruskal)
 {