From 52bf1cd765f52941cc2ea2394832d2f6a5ae622e Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Sun, 13 Jun 2021 10:24:15 -0400
Subject: [PATCH] add cmake command to run GTSAM python tests

---
 python/CMakeLists.txt | 10 ++++++++++
 python/README.md      |  8 ++------
 2 files changed, 12 insertions(+), 6 deletions(-)

diff --git a/python/CMakeLists.txt b/python/CMakeLists.txt
index fb4d89148..5f51368e6 100644
--- a/python/CMakeLists.txt
+++ b/python/CMakeLists.txt
@@ -142,3 +142,13 @@ add_custom_target(${GTSAM_PYTHON_INSTALL_TARGET}
         COMMAND ${PYTHON_EXECUTABLE} ${GTSAM_PYTHON_BUILD_DIRECTORY}/setup.py install
         DEPENDS ${GTSAM_PYTHON_DEPENDENCIES}
         WORKING_DIRECTORY ${GTSAM_PYTHON_BUILD_DIRECTORY})
+
+# Custom make command to run all GTSAM Python tests
+add_custom_target(
+        python-test
+        COMMAND
+          ${CMAKE_COMMAND} -E env # add package to python path so no need to install
+          "PYTHONPATH=${GTSAM_PYTHON_BUILD_DIRECTORY}/$ENV{PYTHONPATH}"
+          ${PYTHON_EXECUTABLE} -m unittest discover
+          DEPENDS ${GTSAM_PYTHON_DEPENDENCIES}
+        WORKING_DIRECTORY ${GTSAM_PYTHON_BUILD_DIRECTORY}/gtsam/tests)
diff --git a/python/README.md b/python/README.md
index c485d12bd..ec9808ce0 100644
--- a/python/README.md
+++ b/python/README.md
@@ -35,12 +35,8 @@ For instructions on updating the version of the [wrap library](https://github.co
 ## Unit Tests
 
 The Python toolbox also has a small set of unit tests located in the
-test directory. To run them:
-
-  ```bash
-  cd <GTSAM_SOURCE_DIRECTORY>/python/gtsam/tests
-  python -m unittest discover
-  ```
+test directory.
+To run them, use `make python-test`.
 
 ## Utils