feat: tests for bib filter

Author: Dariush Wahdany

tests/test_bib_processor.py

@@ -0,0 +1,133 @@
+"""Test cases for BibTeX processing functionality."""
+
+import pytest
+from pathlib import Path
+from tex_minify.bib_processor import extract_citations, filter_bib_file
+
+
+def test_extract_single_citation():
+    content = r"""
+    This is a test document with a single citation \cite{key1}.
+    """
+    citations = extract_citations(content)
+    assert citations == {"key1"}
+
+
+def test_extract_multiple_citations_single_cite():
+    content = r"""
+    This is a test document with multiple citations \cite{key1,key2, key3}.
+    """
+    citations = extract_citations(content)
+    assert citations == {"key1", "key2", "key3"}
+
+
+def test_extract_multiple_cite_commands():
+    content = r"""
+    Different types of citations:
+    \cite{key1}
+    \citep{key2}
+    \citet{key3}
+    """
+    citations = extract_citations(content)
+    assert citations == {"key1", "key2", "key3"}
+
+
+def test_extract_duplicate_citations():
+    content = r"""
+    Duplicate citations:
+    \cite{key1}
+    \cite{key2,key1}
+    \citep{key1,key3}
+    """
+    citations = extract_citations(content)
+    assert citations == {"key1", "key2", "key3"}
+
+
+def test_extract_no_citations():
+    content = "This is a test document with no citations."
+    citations = extract_citations(content)
+    assert citations == set()
+
+
+def test_filter_bib_file(tmp_path):
+    # Create a temporary BibTeX file
+    bib_content = """
+@article{key1,
+    title = {Title 1},
+    author = {Author 1},
+}
+
+@book{key2,
+    title = {Title 2},
+    author = {Author 2},
+}
+
+@inproceedings{key3,
+    title = {Title 3},
+    author = {Author 3},
+}
+"""
+    bib_file = tmp_path / "test.bib"
+    bib_file.write_text(bib_content)
+
+    # Test filtering with different citation sets
+    used_citations = {"key1", "key3"}
+    filtered = filter_bib_file(bib_file, used_citations)
+    
+    # Check that only key1 and key3 entries are present
+    assert "@article{key1," in filtered
+    assert "@inproceedings{key3," in filtered
+    assert "@book{key2," not in filtered
+
+
+def test_filter_bib_file_no_matches(tmp_path):
+    bib_content = """
+@article{key1,
+    title = {Title 1},
+    author = {Author 1},
+}
+"""
+    bib_file = tmp_path / "test.bib"
+    bib_file.write_text(bib_content)
+
+    used_citations = {"key2"}
+    filtered = filter_bib_file(bib_file, used_citations)
+    assert filtered.strip() == ""
+
+
+def test_filter_bib_file_empty_citations(tmp_path):
+    bib_content = """
+@article{key1,
+    title = {Title 1},
+    author = {Author 1},
+}
+"""
+    bib_file = tmp_path / "test.bib"
+    bib_file.write_text(bib_content)
+
+    used_citations = set()
+    filtered = filter_bib_file(bib_file, used_citations)
+    assert filtered.strip() == ""
+
+
+def test_filter_bib_file_with_comments(tmp_path):
+    bib_content = """
+% This is a comment
+@article{key1,
+    title = {Title 1},
+    author = {Author 1},
+}
+
+% Another comment
+@book{key2,
+    title = {Title 2},
+    author = {Author 2},
+}
+"""
+    bib_file = tmp_path / "test.bib"
+    bib_file.write_text(bib_content)
+
+    used_citations = {"key1"}
+    filtered = filter_bib_file(bib_file, used_citations)
+    assert "@article{key1," in filtered
+    assert "@book{key2," not in filtered 

tests/test_cli.py

@@ -0,0 +1,119 @@
+"""Test cases for CLI functionality."""
+
+import pytest
+from click.testing import CliRunner
+from pathlib import Path
+from tex_minify.cli import cli
+
+
+def test_minify_command(tmp_path):
+    runner = CliRunner()
+    
+    # Create a test input file
+    input_file = tmp_path / "test.tex"
+    input_file.write_text(r"\input{chapter1}")
+    
+    # Create the input file that will be referenced
+    chapter_file = tmp_path / "chapter1.tex"
+    chapter_file.write_text("Chapter 1 content")
+    
+    # Test with output to file
+    output_file = tmp_path / "output.tex"
+    result = runner.invoke(cli, ["minify", str(input_file), "-o", str(output_file)])
+    assert result.exit_code == 0
+    assert output_file.read_text() == "Chapter 1 content"
+    
+    # Test with output to stdout
+    result = runner.invoke(cli, ["minify", str(input_file)])
+    assert result.exit_code == 0
+    assert result.output == "Chapter 1 content\n"
+
+
+def test_filter_bib_command(tmp_path):
+    runner = CliRunner()
+    
+    # Create test input files
+    tex_file = tmp_path / "test.tex"
+    tex_file.write_text(r"""
+    This is a test document.
+    \cite{key1,key2}
+    \citep{key3}
+    """)
+    
+    bib_file = tmp_path / "test.bib"
+    bib_file.write_text("""
+@article{key1,
+    title = {Title 1},
+    author = {Author 1},
+}
+
+@book{key2,
+    title = {Title 2},
+    author = {Author 2},
+}
+
+@inproceedings{key3,
+    title = {Title 3},
+    author = {Author 3},
+}
+
+@misc{key4,
+    title = {Title 4},
+    author = {Author 4},
+}
+""")
+    
+    # Test with output to file
+    output_file = tmp_path / "filtered.bib"
+    result = runner.invoke(cli, ["filter-bib", str(tex_file), str(bib_file), "-o", str(output_file)])
+    assert result.exit_code == 0
+    
+    filtered_content = output_file.read_text()
+    assert "@article{key1," in filtered_content
+    assert "@book{key2," in filtered_content
+    assert "@inproceedings{key3," in filtered_content
+    assert "@misc{key4," not in filtered_content
+    
+    # Test with output to stdout
+    result = runner.invoke(cli, ["filter-bib", str(tex_file), str(bib_file)])
+    assert result.exit_code == 0
+    assert "@article{key1," in result.output
+    assert "@book{key2," in result.output
+    assert "@inproceedings{key3," in result.output
+    assert "@misc{key4," not in result.output
+
+
+def test_filter_bib_command_no_citations(tmp_path):
+    runner = CliRunner()
+    
+    # Create test input files with no citations
+    tex_file = tmp_path / "test.tex"
+    tex_file.write_text("This is a test document with no citations.")
+    
+    bib_file = tmp_path / "test.bib"
+    bib_file.write_text("""
+@article{key1,
+    title = {Title 1},
+    author = {Author 1},
+}
+""")
+    
+    # Test with output to file
+    output_file = tmp_path / "filtered.bib"
+    result = runner.invoke(cli, ["filter-bib", str(tex_file), str(bib_file), "-o", str(output_file)])
+    assert result.exit_code == 0
+    assert output_file.read_text().strip() == ""
+
+
+def test_filter_bib_command_missing_files(tmp_path):
+    runner = CliRunner()
+    
+    # Test with non-existent TeX file
+    result = runner.invoke(cli, ["filter-bib", "nonexistent.tex", "test.bib"])
+    assert result.exit_code != 0
+    
+    # Test with non-existent BibTeX file
+    tex_file = tmp_path / "test.tex"
+    tex_file.write_text(r"\cite{key1}")
+    result = runner.invoke(cli, ["filter-bib", str(tex_file), "nonexistent.bib"])
+    assert result.exit_code != 0