From 825a94f7f0574075e323875c1d26206e88742024 Mon Sep 17 00:00:00 2001 From: "Paul Gauthier (aider)" Date: Tue, 27 Aug 2024 14:05:10 -0700 Subject: [PATCH] fix: strip whitespace from shell command in test assertion --- tests/basic/test_coder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/basic/test_coder.py b/tests/basic/test_coder.py index 54b362a6d..79d5650c2 100644 --- a/tests/basic/test_coder.py +++ b/tests/basic/test_coder.py @@ -773,7 +773,7 @@ This command will print 'Hello, World!' to the console.""" # Check if the shell command was added to the list self.assertEqual(len(coder.shell_commands), 1) - self.assertEqual(coder.shell_commands[0], 'echo "Hello, World!"') + self.assertEqual(coder.shell_commands[0].strip(), 'echo "Hello, World!"') # Check if handle_shell_commands was called with the correct argument coder.handle_shell_commands.assert_called_once_with('echo "Hello, World!"', ANY)