Merge pull request #202 from ayushbaid/patch-1

Fixing the test for merging two sets in the DSF
borglab · Jan 11, 2020 · 4169442 · 4169442
2 parents 7ebfdad + 762788e
commit 4169442
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cython/gtsam/tests/test_dsf_map.py b/cython/gtsam/tests/test_dsf_map.py
@@ -30,8 +30,10 @@ def key(index_pair):
         pair1 = gtsam.IndexPair(1, 18)
         self.assertEqual(key(dsf.find(pair1)), key(pair1))
         pair2 = gtsam.IndexPair(2, 2)
+
+        # testing the merge feature of dsf
         dsf.merge(pair1, pair2)
-        self.assertTrue(dsf.find(pair1), dsf.find(pair1))
+        self.assertEquals(key(dsf.find(pair1)), key(dsf.find(pair2)))
 
 
 if __name__ == '__main__':