From 7b2d1c0ffd0b874e280292b926f328a61cb31e2c Mon Sep 17 00:00:00 2001
From: Simon Willison <swillison@gmail.com>
Date: Tue, 25 Oct 2022 12:23:20 -0700
Subject: [PATCH] Update tests for Python 3.11, closes #502

---
 tests/test_cli.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/tests/test_cli.py b/tests/test_cli.py
index b8df8d6..20fe99e 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -737,10 +737,7 @@ def test_query_invalid_function(db_path):
         cli.cli, [db_path, "select bad()", "--functions", "def invalid_python"]
     )
     assert result.exit_code == 1
-    assert (
-        result.output.strip()
-        == "Error: Error in functions definition: invalid syntax (<string>, line 1)"
-    )
+    assert result.output.startswith("Error: Error in functions definition:")
 
 
 TEST_FUNCTIONS = """