From b0d4f3c629d2ba3c35dfa3a18f67bfa48614d1de Mon Sep 17 00:00:00 2001
From: "Edward Z. Yang" <ezyang@mit.edu>
Date: Sun, 27 Apr 2025 14:31:49 -0400
Subject: [PATCH] Update

[ghstack-poisoned]
---
 e2e/test_run_command.py | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/e2e/test_run_command.py b/e2e/test_run_command.py
index d563ff8..dfca1d1 100644
--- a/e2e/test_run_command.py
+++ b/e2e/test_run_command.py
@@ -60,7 +60,16 @@ def test_run_command_with_args(project_dir):
         f.write("test content")
 
     result = subprocess.run(
-        [sys.executable, "-m", "codemcp", "run", "list", test_file, "--path", project_dir],
+        [
+            sys.executable,
+            "-m",
+            "codemcp",
+            "run",
+            "list",
+            test_file,
+            "--path",
+            project_dir,
+        ],
         capture_output=True,
         text=True,
         check=True,
@@ -72,7 +81,15 @@ def test_run_command_error_exit_code(project_dir):
     """Test that error exit codes from the command are propagated."""
     # This should return a non-zero exit code
     process = subprocess.run(
-        [sys.executable, "-m", "codemcp", "run", "exit_with_error", "--path", project_dir],
+        [
+            sys.executable,
+            "-m",
+            "codemcp",
+            "run",
+            "exit_with_error",
+            "--path",
+            project_dir,
+        ],
         capture_output=True,
         text=True,
         check=False,