Skip to content

Commit 39a5a82

Browse files
committed
add top_k=2 at test_code
1 parent 1899f3e commit 39a5a82

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/autorag/nodes/passageaugmenter/test_recursive_chunk.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,12 @@ def test_recursive_chunk():
1111

1212

1313
def test_recursive_chunk_node():
14-
result_df = recursive_chunk(project_dir=project_dir, previous_result=previous_result, top_k=3)
14+
result_df = recursive_chunk(project_dir=project_dir, previous_result=previous_result, top_k=2)
1515
contents = result_df["retrieved_contents"].tolist()
1616
ids = result_df["retrieved_ids"].tolist()
1717
scores = result_df["retrieve_scores"].tolist()
1818
assert len(contents) == len(ids) == len(scores) == 2
19-
assert len(contents[0]) == len(ids[0]) == len(scores[0]) == 3
19+
assert len(contents[0]) == len(ids[0]) == len(scores[0]) == 2
2020
for content_list, id_list, score_list in zip(contents, ids, scores):
2121
for i, (content, _id, score) in enumerate(zip(content_list, id_list, score_list)):
2222
assert isinstance(content, str)

0 commit comments

Comments
 (0)