diff --git a/scripts/no_bool_in_generic.py b/scripts/no_bool_in_generic.py index c483d221a4266..15bc2247469c7 100644 --- a/scripts/no_bool_in_generic.py +++ b/scripts/no_bool_in_generic.py @@ -65,7 +65,7 @@ def replace_bool_with_bool_t(to_replace, content: str) -> str: + replaced_line[col_offset + 4 :] ) new_lines.append(replaced_line) - return "\n".join(new_lines) + return "\n".join(new_lines) + "\n" def check_for_bool_in_generic(content: str) -> tuple[bool, str]: diff --git a/scripts/tests/test_no_bool_in_generic.py b/scripts/tests/test_no_bool_in_generic.py index 0bc91c5d1cf1e..a57612548dcc5 100644 --- a/scripts/tests/test_no_bool_in_generic.py +++ b/scripts/tests/test_no_bool_in_generic.py @@ -1,7 +1,7 @@ from scripts.no_bool_in_generic import check_for_bool_in_generic -BAD_FILE = "def foo(a: bool) -> bool:\n return bool(0)" -GOOD_FILE = "def foo(a: bool_t) -> bool_t:\n return bool(0)" +BAD_FILE = "def foo(a: bool) -> bool:\n return bool(0)\n" +GOOD_FILE = "def foo(a: bool_t) -> bool_t:\n return bool(0)\n" def test_bad_file_with_replace():