From 55044ba23152bd84c1e0803a7a07d3142d3738f0 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Fri, 2 Aug 2024 12:56:53 -0700
Subject: [PATCH] Add make test_watch_all command

---
 libs/checkpoint/Makefile       |  3 +++
 libs/checkpoint/poetry.lock    | 29 +++++++++--------------------
 libs/checkpoint/pyproject.toml |  8 +++++++-
 libs/langgraph/Makefile        |  3 +++
 4 files changed, 22 insertions(+), 21 deletions(-)

diff --git a/libs/checkpoint/Makefile b/libs/checkpoint/Makefile
index 93d1b4ca0..ccbf25719 100644
--- a/libs/checkpoint/Makefile
+++ b/libs/checkpoint/Makefile
@@ -7,6 +7,9 @@
 test:
 	poetry run pytest tests
 
+test_watch:
+	poetry run ptw .
+
 ######################
 # LINTING AND FORMATTING
 ######################
diff --git a/libs/checkpoint/poetry.lock b/libs/checkpoint/poetry.lock
index d8cccfcc1..ba1dd1ccd 100644
--- a/libs/checkpoint/poetry.lock
+++ b/libs/checkpoint/poetry.lock
@@ -182,16 +182,6 @@ files = [
 marshmallow = ">=3.18.0,<4.0.0"
 typing-inspect = ">=0.4.0,<1"
 
-[[package]]
-name = "docopt"
-version = "0.6.2"
-description = "Pythonic argument parser, that will make you smile"
-optional = false
-python-versions = "*"
-files = [
-    {file = "docopt-0.6.2.tar.gz", hash = "sha256:49b3a825280bd66b3aa83585ef59c4a8c82f2c8a522dbe754a8bc8d08c85c491"},
-]
-
 [[package]]
 name = "exceptiongroup"
 version = "1.2.2"
@@ -638,20 +628,19 @@ pytest = ">=6.2.5"
 dev = ["pre-commit", "pytest-asyncio", "tox"]
 
 [[package]]
-name = "pytest-watch"
-version = "4.2.0"
-description = "Local continuous test runner with pytest and watchdog."
+name = "pytest-watcher"
+version = "0.4.2"
+description = "Automatically rerun your tests on file modifications"
 optional = false
-python-versions = "*"
+python-versions = "<4.0.0,>=3.7.0"
 files = [
-    {file = "pytest-watch-4.2.0.tar.gz", hash = "sha256:06136f03d5b361718b8d0d234042f7b2f203910d8568f63df2f866b547b3d4b9"},
+    {file = "pytest_watcher-0.4.2-py3-none-any.whl", hash = "sha256:a43949ba67dd8d7e1fd0de5eea44a999081f0aec9f93b4e744264b4c6a3d9bbe"},
+    {file = "pytest_watcher-0.4.2.tar.gz", hash = "sha256:7b292f025ca19617cd7567c228c6187b5087f2da9e4d2cf6e144e5764a0471b0"},
 ]
 
 [package.dependencies]
-colorama = ">=0.3.3"
-docopt = ">=0.4.0"
-pytest = ">=2.6.4"
-watchdog = ">=0.6.0"
+tomli = {version = ">=2.0.1,<3.0.0", markers = "python_version < \"3.11\""}
+watchdog = ">=2.0.0"
 
 [[package]]
 name = "pyyaml"
@@ -876,4 +865,4 @@ watchmedo = ["PyYAML (>=3.10)"]
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.9.0,<4.0"
-content-hash = "16d70d6ae432212abef4f8a0ba9411e593ff63b9106695d8714589fa0cabf97f"
+content-hash = "d035f669f68dca5d9a30bd21cafe31195df4c175d8698ae4d4d19a11e4035071"
diff --git a/libs/checkpoint/pyproject.toml b/libs/checkpoint/pyproject.toml
index d258d12f0..4a290eee4 100644
--- a/libs/checkpoint/pyproject.toml
+++ b/libs/checkpoint/pyproject.toml
@@ -18,7 +18,7 @@ codespell = "^2.2.0"
 pytest = "^7.2.1"
 pytest-asyncio = "^0.21.1"
 pytest-mock = "^3.11.1"
-pytest-watch = "^4.2.0"
+pytest-watcher = "^0.4.1"
 mypy = "^1.10.0"
 dataclasses-json = "^0.6.7"
 aiosqlite = "^0.20.0"
@@ -47,3 +47,9 @@ lint.select = [
   "I",  # isort
 ]
 lint.ignore = ["E501", "B008", "UP007", "UP006"]
+
+[tool.pytest-watcher]
+now = true
+delay = 0.1
+runner_args = ["--ff", "-v", "--tb", "short"]
+patterns = ["*.py"]
diff --git a/libs/langgraph/Makefile b/libs/langgraph/Makefile
index 8403fbf33..80fa00354 100644
--- a/libs/langgraph/Makefile
+++ b/libs/langgraph/Makefile
@@ -20,6 +20,9 @@ test:
 test_watch:
 	poetry run ptw .
 
+test_watch_all:
+	npx concurrently -n langgraph,checkpoint "make test_watch" "make -C ../checkpoint test_watch"
+
 ######################
 # LINTING AND FORMATTING
 ######################