From f50f963e57d45caebf1d8ad8856b1e110f5d9d84 Mon Sep 17 00:00:00 2001
From: Frank Dellaert <dellaert@gmail.com>
Date: Wed, 27 Oct 2021 13:44:54 -0400
Subject: [PATCH] Add main

---
 python/gtsam/tests/test_DiscreteFactorGraph.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/python/gtsam/tests/test_DiscreteFactorGraph.py b/python/gtsam/tests/test_DiscreteFactorGraph.py
index afc6630bd..9ed7cc010 100644
--- a/python/gtsam/tests/test_DiscreteFactorGraph.py
+++ b/python/gtsam/tests/test_DiscreteFactorGraph.py
@@ -336,10 +336,6 @@ class TestDiscreteFactorGraph(GtsamTestCase):
 # }
 # #endif
 
-# /* ************************************************************************* */
-# int main() {
-# TestResult tr
-# return TestRegistry::runAllTests(tr)
-# }
-# /* ************************************************************************* */
 
+if __name__ == "__main__":
+    unittest.main()