Fixed failing shell tests (#3045)

kuzudb · Mar 14, 2024 · 1b6f741 · 1b6f741
1 parent 930ba45
commit 1b6f741
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/shell/test/test_shell_basics.py b/tools/shell/test/test_shell_basics.py
@@ -72,7 +72,7 @@ def test_multi_queries_one_line(temp_db) -> None:
     result.check_stdout("databases rule")
     result.check_stdout(
         [
-            "Error: Parser exception: mismatched input '<EOF>' expecting {CALL, COMMENT, COPY, EXPORT, DROP, ALTER, BEGIN, COMMIT, COMMIT_SKIP_CHECKPOINT, ROLLBACK, ROLLBACK_SKIP_CHECKPOINT, INSTALL, LOAD, OPTIONAL, MATCH, UNWIND, CREATE, MERGE, SET, DETACH, DELETE, WITH, RETURN} (line: 1, offset: 6)",
+            "Error: Parser exception: mismatched input '<EOF>' expecting {CALL, COMMENT_, COPY, EXPORT, IMPORT, DROP, ALTER, BEGIN, COMMIT, COMMIT_SKIP_CHECKPOINT, ROLLBACK, ROLLBACK_SKIP_CHECKPOINT, INSTALL, LOAD, OPTIONAL, MATCH, UNWIND, CREATE, MERGE, SET, DETACH, DELETE, WITH, RETURN} (line: 1, offset: 6)",
             '"      "',
         ],
     )
@@ -85,7 +85,7 @@ def test_multi_queries_one_line(temp_db) -> None:
             "Error: Parser exception: Invalid input < S>: expected rule ku_Statements (line: 1, offset: 24)",
             '"RETURN "databases rule" S a"',
             "                         ^",
-            "Error: Parser exception: mismatched input '<EOF>' expecting {CALL, COMMENT, COPY, EXPORT, DROP, ALTER, BEGIN, COMMIT, COMMIT_SKIP_CHECKPOINT, ROLLBACK, ROLLBACK_SKIP_CHECKPOINT, INSTALL, LOAD, OPTIONAL, MATCH, UNWIND, CREATE, MERGE, SET, DETACH, DELETE, WITH, RETURN} (line: 1, offset: 6)",
+            "Error: Parser exception: mismatched input '<EOF>' expecting {CALL, COMMENT_, COPY, EXPORT, IMPORT, DROP, ALTER, BEGIN, COMMIT, COMMIT_SKIP_CHECKPOINT, ROLLBACK, ROLLBACK_SKIP_CHECKPOINT, INSTALL, LOAD, OPTIONAL, MATCH, UNWIND, CREATE, MERGE, SET, DETACH, DELETE, WITH, RETURN} (line: 1, offset: 6)",
             '"      "',
         ],
     )