Skip to content

Commit 6714466

Browse files
author
WangGLJoseph
committed
Merge branch 'pytest-ci' into pytest-unit
2 parents 77dba68 + d06e4e7 commit 6714466

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

app/data_structures.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def __init__(
5151
)
5252

5353
def __str__(self):
54-
return f"Call function `{self.func_name}` with arguments {self.arg_values}."
54+
return f"FunctionCallIntent(func_name={self.func_name}, arguments={self.arg_values})"
5555

5656
def to_dict(self):
5757
return {"func_name": self.func_name, "arguments": self.arg_values}

app/search/search_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -340,7 +340,7 @@ def get_code_region_around_line(
340340

341341
# start and end should also be 1-based valid line numbers
342342
start = max(1, line_no - window_size)
343-
end = min(len(file_content), line_no + window_size)
343+
end = min(len(file_content) + 1, line_no + window_size)
344344
snippet = ""
345345
for i in range(start, end):
346346
if with_lineno:

test/app/search/test_search_utils.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -664,14 +664,15 @@ def test_get_code_region_around_line_edge_of_file(tmp_path):
664664
temp_file.write_text(file_content)
665665

666666
# Choose a line number near the end.
667-
# For line_no = 8, window_size = 5, start = max(1, 8-5)=3, end = min(8, 8+5)=8.
668-
# Loop runs for i in range(3, 8) → lines 3 to 7.
667+
# For line_no = 8, window_size = 5, start = max(1, 8-5)=3, end = min(9, 8+5)=9.
668+
# Loop runs for i in range(3, 9) → lines 3 to 8.
669669
expected = (
670670
"3 edge 3\n"
671671
"4 edge 4\n"
672672
"5 edge 5\n"
673673
"6 edge 6\n"
674674
"7 edge 7\n"
675+
"8 edge 8\n"
675676
)
676677
result = get_code_region_around_line(str(temp_file), 8, window_size=5, with_lineno=True)
677678
assert result == expected, f"Expected:\n{expected}\nGot:\n{result}"

0 commit comments

Comments
 (0)