arxiv-to-prompt 0.2.0__tar.gz → 0.2.1__tar.gz

{arxiv_to_prompt-0.2.0/src/arxiv_to_prompt.egg-info → arxiv_to_prompt-0.2.1}/PKG-INFO

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

{arxiv_to_prompt-0.2.0 → arxiv_to_prompt-0.2.1}/pyproject.toml

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

{arxiv_to_prompt-0.2.0 → arxiv_to_prompt-0.2.1}/src/arxiv_to_prompt/core.py

@@ -192,7 +192,8 @@ def flatten_tex(directory: str, main_file: str) -> str:
                 
                 # Process the command normally
                 input_file = match.group(1)
-                if not input_file.endswith('.tex'):
+                # Only add .tex extension if the file has no extension at all
+                if not os.path.splitext(input_file)[1]:
                     input_file += '.tex'
                 input_path = os.path.join(directory, input_file)
                 return process_file(input_path, processed_files)

{arxiv_to_prompt-0.2.0 → arxiv_to_prompt-0.2.1/src/arxiv_to_prompt.egg-info}/PKG-INFO

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

{arxiv_to_prompt-0.2.0 → arxiv_to_prompt-0.2.1}/tests/test_core.py

@@ -228,3 +228,46 @@ def test_process_latex_with_appendix_removal(sample_arxiv_id, temp_cache_dir):
     
     # Check that appendix was removed (if it existed)
     assert "\\appendix" not in result
+
+
+def test_input_file_extensions(temp_cache_dir):
+    """Test that input files with existing extensions are not modified."""
+    # Create test directory and files
+    tex_dir = temp_cache_dir / "test_extensions"
+    tex_dir.mkdir(parents=True)
+    
+    # Create main file with various input commands
+    main_file = tex_dir / "main.tex"
+    main_content = """\\documentclass{article}
+\\begin{document}
+\\input{chapter1}
+\\input{main.bbl}
+\\input{mystyle.sty}
+\\input{config.cls}
+\\input{already.tex}
+\\end{document}
+"""
+    main_file.write_text(main_content)
+    
+    # Create the files that should be included
+    files_to_create = [
+        ("chapter1.tex", "Chapter 1 content"),
+        ("main.bbl", "Bibliography content"),
+        ("mystyle.sty", "Style content"),
+        ("config.cls", "Class content"),
+        ("already.tex", "Already tex content"),
+    ]
+    
+    for filename, content in files_to_create:
+        file_path = tex_dir / filename
+        file_path.write_text(content)
+    
+    # Run the flatten_tex function
+    result = flatten_tex(str(tex_dir), "main.tex")
+    
+    # Check that all files were included correctly
+    assert "Chapter 1 content" in result
+    assert "Bibliography content" in result
+    assert "Style content" in result
+    assert "Class content" in result
+    assert "Already tex content" in result