arxiv-to-prompt 0.4.0__tar.gz → 0.4.1__tar.gz

{arxiv_to_prompt-0.4.0/src/arxiv_to_prompt.egg-info → arxiv_to_prompt-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arxiv-to-prompt
-Version: 0.4.0
+Version: 0.4.1
 Summary: transform arXiv papers into a single latex prompt for LLMs
 Author: Takashi Ishida
 License: MIT

{arxiv_to_prompt-0.4.0 → arxiv_to_prompt-0.4.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "arxiv-to-prompt"
-version = "0.4.0"
+version = "0.4.1"
 description = "transform arXiv papers into a single latex prompt for LLMs"
 readme = "README.md"
 authors = [{ name = "Takashi Ishida" }]

{arxiv_to_prompt-0.4.0 → arxiv_to_prompt-0.4.1}/src/arxiv_to_prompt/core.py

@@ -131,6 +131,8 @@ def find_main_tex(directory: str) -> Optional[str]:
 
 def remove_comments_from_lines(text: str) -> str:
     """Remove LaTeX comments while preserving newlines."""
+    # Remove \iffalse...\fi blocks (commonly used to comment out large sections)
+    text = re.sub(r'\\iffalse\b.*?\\fi\b', '', text, flags=re.DOTALL)
     lines = text.split('\n')
     result = []
     for line in lines:

{arxiv_to_prompt-0.4.0 → arxiv_to_prompt-0.4.1/src/arxiv_to_prompt.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arxiv-to-prompt
-Version: 0.4.0
+Version: 0.4.1
 Summary: transform arXiv papers into a single latex prompt for LLMs
 Author: Takashi Ishida
 License: MIT

{arxiv_to_prompt-0.4.0 → arxiv_to_prompt-0.4.1}/tests/test_core.py

@@ -109,11 +109,29 @@ def test_remove_comments_from_lines():
         ("Multiple % comments % here", "Multiple"),
         ("Line with both \\% and % real comment", "Line with both \\% and"),
     ]
-    
+
     for input_text, expected in test_cases:
         assert remove_comments_from_lines(input_text).rstrip() == expected
 
 
+def test_remove_iffalse_blocks():
+    """Test removal of \\iffalse...\\fi blocks."""
+    # Single line
+    assert remove_comments_from_lines("before \\iffalse hidden \\fi after") == "before  after"
+
+    # Multi-line block
+    input_text = "before\n\\iffalse\nhidden\ncontent\n\\fi\nafter"
+    result = remove_comments_from_lines(input_text)
+    assert "hidden" not in result
+    assert "before" in result
+    assert "after" in result
+
+    # Multiple blocks
+    input_text = "a \\iffalse x \\fi b \\iffalse y \\fi c"
+    result = remove_comments_from_lines(input_text)
+    assert result == "a  b  c"
+
+
 def test_find_main_tex(temp_cache_dir):
     """Test finding the main tex file."""
     # Create test files