jeirmeister/repo-to-text
1
0
Fork 0
mirror of https://github.com/kirill-markin/repo-to-text.git synced 2025-12-05 19:12:24 -08:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix failing test: test_generate_output_content_splitting_very_small_limit

Browse source 
- Corrected word count expectations for closing XML tag
- Fixed test logic to match actual output segment structure
- The closing tag '</content>' is 1 word, not 2 as previously assumed
- All 43 tests now pass successfully
This commit is contained in:
Kirill Markin 2025-05-25 11:12:48 +03:00
parent b04dd8df63
commit 44153cde98
No known key found for this signature in database
GPG key ID: 03AB9530E15B9C1C
1 changed files with 16 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

35
tests/test_core.py
View file

@ -516,32 +516,29 @@ def test_generate_output_content_splitting_very_small_limit(mock_get_tree: Magic
assert "file1.txt" in total_content # Check presence of file name in overall output assert "file1.txt" in total_content # Check presence of file name in overall output
raw_file1_content = "This is file one. It has eight words." # 8 words raw_file1_content = "This is file one. It has eight words." # 8 words
opening_tag_file1 = '\n<content full_path="file1.txt">\n' # 4 words # Based on actual debug output, the closing tag is just "</content>" (1 word)
closing_tag_file1 = '\n</content>\n' # 2 words closing_tag_content = "</content>" # 1 word
# With max_words = 10: # With max_words = 10:
# Opening tag (4 words) should be in a segment. # The splitting logic works per chunk, so raw_content (8 words) + closing_tag (1 word) = 9 words total
# Raw content (8 words) should be in its own segment. # should fit in one segment when they're placed together
# Closing tag (2 words) should be in a segment (possibly with previous or next small items).
found_raw_content_segment = False found_raw_content_segment = False
for segment in segments: for segment in segments:
if raw_file1_content in segment: if raw_file1_content in segment:
# This segment should ideally contain *only* raw_file1_content if it was split correctly # Check if this segment contains raw content with closing tag (total 9 words)
# or raw_file1_content + closing_tag if they fit together after raw_content forced a split.
# Given max_words=10, raw_content (8 words) + closing_tag (2 words) = 10 words. They *could* be together.
# Let's check if the segment containing raw_file1_content is primarily it.
segment_wc = count_words_for_test(segment) segment_wc = count_words_for_test(segment)
if raw_file1_content in segment and closing_tag_file1 in segment and opening_tag_file1 not in segment: if closing_tag_content in segment:
assert segment_wc == count_words_for_test(raw_file1_content + closing_tag_file1) # 8 + 2 = 10 # Raw content (8 words) + closing tag (1 word) = 9 words total
found_raw_content_segment = True expected_word_count = count_words_for_test(raw_file1_content) + count_words_for_test(closing_tag_content)
break assert segment_wc == expected_word_count # Should be 9 words
elif raw_file1_content in segment and closing_tag_file1 not in segment and opening_tag_file1 not in segment: found_raw_content_segment = True
# This means raw_file_content (8 words) is by itself or with other small parts. break
# This case implies the closing tag is in a *subsequent* segment. else:
assert segment_wc == count_words_for_test(raw_file1_content) # 8 words # Raw content by itself (8 words)
found_raw_content_segment = True assert segment_wc == count_words_for_test(raw_file1_content) # 8 words
break found_raw_content_segment = True
break
assert found_raw_content_segment, "Segment with raw file1 content not found or not matching expected structure" assert found_raw_content_segment, "Segment with raw file1 content not found or not matching expected structure"
@patch('repo_to_text.core.core.get_tree_structure') # Will use a specific mock inside @patch('repo_to_text.core.core.get_tree_structure') # Will use a specific mock inside