From 0d4edee47658dec59fb720518b85692c023858b3 Mon Sep 17 00:00:00 2001
From: tserg <8017125+tserg@users.noreply.github.com>
Date: Fri, 4 Oct 2024 17:57:50 +0800
Subject: [PATCH] fix return value of pre_parse for tests

---
 tests/functional/grammar/test_grammar.py | 2 +-
 tests/unit/ast/test_pre_parser.py        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/functional/grammar/test_grammar.py b/tests/functional/grammar/test_grammar.py
index 2af5385b3d..e95578e7e3 100644
--- a/tests/functional/grammar/test_grammar.py
+++ b/tests/functional/grammar/test_grammar.py
@@ -102,6 +102,6 @@ def has_no_docstrings(c):
     max_examples=500, suppress_health_check=[HealthCheck.too_slow, HealthCheck.filter_too_much]
 )
 def test_grammar_bruteforce(code):
-    _, _, _, reformatted_code = pre_parse(code + "\n")
+    _, _, _, _, reformatted_code = pre_parse(code + "\n")
     tree = parse_to_ast(reformatted_code)
     assert isinstance(tree, Module)
diff --git a/tests/unit/ast/test_pre_parser.py b/tests/unit/ast/test_pre_parser.py
index da7d72b8ec..dda0cdeb39 100644
--- a/tests/unit/ast/test_pre_parser.py
+++ b/tests/unit/ast/test_pre_parser.py
@@ -174,7 +174,7 @@ def test_prerelease_invalid_version_pragma(file_version, mock_version):
 @pytest.mark.parametrize("code, pre_parse_settings, compiler_data_settings", pragma_examples)
 def test_parse_pragmas(code, pre_parse_settings, compiler_data_settings, mock_version):
     mock_version("0.3.10")
-    settings, _, _, _ = pre_parse(code)
+    settings, _, _, _, _ = pre_parse(code)
 
     assert settings == pre_parse_settings