easybib 0.1.0__tar.gz → 0.2.0__tar.gz

{easybib-0.1.0 → easybib-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: easybib
-Version: 0.1.0
+Version: 0.2.0
 Summary: Automatically fetch BibTeX entries from INSPIRE and ADS for LaTeX projects
 Author-email: Gregory Ashton <gregory.ashton@ligo.org>
 License-Expression: MIT
@@ -9,9 +9,15 @@ Project-URL: Repository, https://github.com/GregoryAshton/easybib
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: requests
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest-cov; extra == "test"
 
 # easybib
 
+[![Tests](https://github.com/GregoryAshton/easybib/actions/workflows/tests.yml/badge.svg)](https://github.com/GregoryAshton/easybib/actions/workflows/tests.yml)
+[![codecov](https://codecov.io/gh/GregoryAshton/easybib/branch/main/graph/badge.svg)](https://codecov.io/gh/GregoryAshton/easybib)
+
 Automatically fetch BibTeX entries from [INSPIRE](https://inspirehep.net/) and [NASA/ADS](https://ui.adsabs.harvard.edu/) for LaTeX projects.
 
 easybib scans your `.tex` files for citation keys, looks them up on INSPIRE and/or ADS, and writes a `.bib` file with the results. It handles INSPIRE texkeys (e.g. `Author:2020abc`) and ADS bibcodes (e.g. `2016PhRvL.116f1102A`).
@@ -26,9 +32,10 @@ pip install easybib
 
 ```bash
 easybib /path/to/latex/project
+easybib paper.tex
 ```
 
-This will scan all `.tex` files in the directory, fetch BibTeX entries, and write them to `references.bib`.
+Pass a directory to scan all `.tex` files recursively, or a single `.tex` file. BibTeX entries are fetched and written to `references.bib`.
 
 ### Options
 
@@ -39,13 +46,17 @@ This will scan all `.tex` files in the directory, fetch BibTeX entries, and writ
 | `-a`, `--max-authors` | Truncate author lists (default: 3, use 0 for no limit) |
 | `-l`, `--list-keys` | List found citation keys and exit (no fetching) |
 | `--fresh` | Ignore existing output file and start from scratch |
+| `--ads-api-key` | ADS API key (overrides `ADS_API_KEY` environment variable) |
 
 ### Examples
 
 ```bash
-# Fetch from INSPIRE (no API key needed)
+# Scan a directory
 easybib ./paper -s inspire
 
+# Scan a single file
+easybib paper.tex
+
 # Use a custom output file
 easybib ./paper -o paper.bib
 
@@ -58,13 +69,19 @@ easybib ./paper -a 0
 
 ### ADS API key
 
-When using ADS as the source (the default), set your API key:
+When using ADS as the source (the default), provide your API key either via the command line:
+
+```bash
+easybib ./paper --ads-api-key your-key-here
+```
+
+Or as an environment variable:
 
 ```bash
 export ADS_API_KEY="your-key-here"
 ```
 
-Get one from https://ui.adsabs.harvard.edu/user/settings/token.
+Get a key from https://ui.adsabs.harvard.edu/user/settings/token.
 
 ## How it works
 

{easybib-0.1.0 → easybib-0.2.0}/README.md

@@ -1,5 +1,8 @@
 # easybib
 
+[![Tests](https://github.com/GregoryAshton/easybib/actions/workflows/tests.yml/badge.svg)](https://github.com/GregoryAshton/easybib/actions/workflows/tests.yml)
+[![codecov](https://codecov.io/gh/GregoryAshton/easybib/branch/main/graph/badge.svg)](https://codecov.io/gh/GregoryAshton/easybib)
+
 Automatically fetch BibTeX entries from [INSPIRE](https://inspirehep.net/) and [NASA/ADS](https://ui.adsabs.harvard.edu/) for LaTeX projects.
 
 easybib scans your `.tex` files for citation keys, looks them up on INSPIRE and/or ADS, and writes a `.bib` file with the results. It handles INSPIRE texkeys (e.g. `Author:2020abc`) and ADS bibcodes (e.g. `2016PhRvL.116f1102A`).
@@ -14,9 +17,10 @@ pip install easybib
 
 ```bash
 easybib /path/to/latex/project
+easybib paper.tex
 ```
 
-This will scan all `.tex` files in the directory, fetch BibTeX entries, and write them to `references.bib`.
+Pass a directory to scan all `.tex` files recursively, or a single `.tex` file. BibTeX entries are fetched and written to `references.bib`.
 
 ### Options
 
@@ -27,13 +31,17 @@ This will scan all `.tex` files in the directory, fetch BibTeX entries, and writ
 | `-a`, `--max-authors` | Truncate author lists (default: 3, use 0 for no limit) |
 | `-l`, `--list-keys` | List found citation keys and exit (no fetching) |
 | `--fresh` | Ignore existing output file and start from scratch |
+| `--ads-api-key` | ADS API key (overrides `ADS_API_KEY` environment variable) |
 
 ### Examples
 
 ```bash
-# Fetch from INSPIRE (no API key needed)
+# Scan a directory
 easybib ./paper -s inspire
 
+# Scan a single file
+easybib paper.tex
+
 # Use a custom output file
 easybib ./paper -o paper.bib
 
@@ -46,13 +54,19 @@ easybib ./paper -a 0
 
 ### ADS API key
 
-When using ADS as the source (the default), set your API key:
+When using ADS as the source (the default), provide your API key either via the command line:
+
+```bash
+easybib ./paper --ads-api-key your-key-here
+```
+
+Or as an environment variable:
 
 ```bash
 export ADS_API_KEY="your-key-here"
 ```
 
-Get one from https://ui.adsabs.harvard.edu/user/settings/token.
+Get a key from https://ui.adsabs.harvard.edu/user/settings/token.
 
 ## How it works
 

{easybib-0.1.0 → easybib-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "easybib"
-version = "0.1.0"
+version = "0.2.0"
 description = "Automatically fetch BibTeX entries from INSPIRE and ADS for LaTeX projects"
 readme = "README.md"
 license = "MIT"
@@ -16,6 +16,9 @@ dependencies = [
     "requests",
 ]
 
+[project.optional-dependencies]
+test = ["pytest", "pytest-cov"]
+
 [project.urls]
 Homepage = "https://github.com/GregoryAshton/easybib"
 Repository = "https://github.com/GregoryAshton/easybib"

{easybib-0.1.0 → easybib-0.2.0}/src/easybib/cli.py

@@ -17,9 +17,9 @@ def main():
     parser = argparse.ArgumentParser(
         description="Extract citations and download BibTeX from NASA/ADS"
     )
-    parser.add_argument("directory", help="Directory containing LaTeX files")
+    parser.add_argument("path", help="LaTeX file or directory containing LaTeX files")
     parser.add_argument(
-        "-o", "--output", default="references.bib", help="Output BibTeX file"
+        "-o", "--output", default="references.bib", help="Output BibTeX file (existing entries are retained)"
     )
     parser.add_argument(
         "-a",
@@ -46,13 +46,21 @@ def main():
         default="ads",
         help="Preferred BibTeX source: 'ads' (default), 'inspire', or 'auto' (based on key format)",
     )
+    parser.add_argument(
+        "--ads-api-key",
+        help="ADS API key (overrides ADS_API_KEY environment variable)",
+    )
     args = parser.parse_args()
 
     # Collect all citation keys
-    tex_dir = Path(args.directory)
+    input_path = Path(args.path)
     all_keys = set()
     all_warnings = []
-    for tex_file in tex_dir.glob("**/*.tex"):
+    if input_path.is_file():
+        tex_files = [input_path]
+    else:
+        tex_files = input_path.glob("**/*.tex")
+    for tex_file in tex_files:
         keys, warnings = extract_cite_keys(tex_file)
         all_keys.update(keys)
         all_warnings.extend(warnings)
@@ -73,7 +81,7 @@ def main():
         return 0
 
     # Check for ADS API key (not required if using --source inspire)
-    api_key = os.getenv("ADS_API_KEY")
+    api_key = args.ads_api_key or os.getenv("ADS_API_KEY")
     if not api_key and args.source != "inspire":
         print("Error: ADS_API_KEY environment variable not set")
         print("Get your API key from: https://ui.adsabs.harvard.edu/user/settings/token")

{easybib-0.1.0 → easybib-0.2.0}/src/easybib.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: easybib
-Version: 0.1.0
+Version: 0.2.0
 Summary: Automatically fetch BibTeX entries from INSPIRE and ADS for LaTeX projects
 Author-email: Gregory Ashton <gregory.ashton@ligo.org>
 License-Expression: MIT
@@ -9,9 +9,15 @@ Project-URL: Repository, https://github.com/GregoryAshton/easybib
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 Requires-Dist: requests
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest-cov; extra == "test"
 
 # easybib
 
+[![Tests](https://github.com/GregoryAshton/easybib/actions/workflows/tests.yml/badge.svg)](https://github.com/GregoryAshton/easybib/actions/workflows/tests.yml)
+[![codecov](https://codecov.io/gh/GregoryAshton/easybib/branch/main/graph/badge.svg)](https://codecov.io/gh/GregoryAshton/easybib)
+
 Automatically fetch BibTeX entries from [INSPIRE](https://inspirehep.net/) and [NASA/ADS](https://ui.adsabs.harvard.edu/) for LaTeX projects.
 
 easybib scans your `.tex` files for citation keys, looks them up on INSPIRE and/or ADS, and writes a `.bib` file with the results. It handles INSPIRE texkeys (e.g. `Author:2020abc`) and ADS bibcodes (e.g. `2016PhRvL.116f1102A`).
@@ -26,9 +32,10 @@ pip install easybib
 
 ```bash
 easybib /path/to/latex/project
+easybib paper.tex
 ```
 
-This will scan all `.tex` files in the directory, fetch BibTeX entries, and write them to `references.bib`.
+Pass a directory to scan all `.tex` files recursively, or a single `.tex` file. BibTeX entries are fetched and written to `references.bib`.
 
 ### Options
 
@@ -39,13 +46,17 @@ This will scan all `.tex` files in the directory, fetch BibTeX entries, and writ
 | `-a`, `--max-authors` | Truncate author lists (default: 3, use 0 for no limit) |
 | `-l`, `--list-keys` | List found citation keys and exit (no fetching) |
 | `--fresh` | Ignore existing output file and start from scratch |
+| `--ads-api-key` | ADS API key (overrides `ADS_API_KEY` environment variable) |
 
 ### Examples
 
 ```bash
-# Fetch from INSPIRE (no API key needed)
+# Scan a directory
 easybib ./paper -s inspire
 
+# Scan a single file
+easybib paper.tex
+
 # Use a custom output file
 easybib ./paper -o paper.bib
 
@@ -58,13 +69,19 @@ easybib ./paper -a 0
 
 ### ADS API key
 
-When using ADS as the source (the default), set your API key:
+When using ADS as the source (the default), provide your API key either via the command line:
+
+```bash
+easybib ./paper --ads-api-key your-key-here
+```
+
+Or as an environment variable:
 
 ```bash
 export ADS_API_KEY="your-key-here"
 ```
 
-Get one from https://ui.adsabs.harvard.edu/user/settings/token.
+Get a key from https://ui.adsabs.harvard.edu/user/settings/token.
 
 ## How it works
 

{easybib-0.1.0 → easybib-0.2.0}/src/easybib.egg-info/SOURCES.txt

@@ -9,4 +9,7 @@ src/easybib.egg-info/SOURCES.txt
 src/easybib.egg-info/dependency_links.txt
 src/easybib.egg-info/entry_points.txt
 src/easybib.egg-info/requires.txt
-src/easybib.egg-info/top_level.txt
+src/easybib.egg-info/top_level.txt
+tests/test_cli.py
+tests/test_core.py
+tests/test_fetch.py

easybib-0.2.0/src/easybib.egg-info/requires.txt

@@ -0,0 +1,5 @@
+requests
+
+[test]
+pytest
+pytest-cov

easybib-0.2.0/tests/test_cli.py

@@ -0,0 +1,101 @@
+"""Tests for easybib CLI."""
+
+import subprocess
+import sys
+from unittest.mock import patch
+
+from easybib.cli import main
+
+
+class TestListKeys:
+    def test_list_keys_single_file(self, tmp_path, capsys):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\cite{Author:2020abc} and \citep{Other:2021xyz}")
+        with patch("sys.argv", ["easybib", str(tex), "--list-keys"]):
+            result = main()
+        assert result == 0
+        captured = capsys.readouterr()
+        assert "Author:2020abc" in captured.out
+        assert "Other:2021xyz" in captured.out
+
+    def test_list_keys_directory(self, tmp_path, capsys):
+        sub = tmp_path / "subdir"
+        sub.mkdir()
+        (tmp_path / "a.tex").write_text(r"\cite{A:2020abc}")
+        (sub / "b.tex").write_text(r"\cite{B:2021xyz}")
+        with patch("sys.argv", ["easybib", str(tmp_path), "--list-keys"]):
+            result = main()
+        assert result == 0
+        captured = capsys.readouterr()
+        assert "A:2020abc" in captured.out
+        assert "B:2021xyz" in captured.out
+
+
+class TestMissingApiKey:
+    def test_error_without_api_key(self, tmp_path, capsys):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\cite{Author:2020abc}")
+        with (
+            patch("sys.argv", ["easybib", str(tex)]),
+            patch.dict("os.environ", {}, clear=True),
+        ):
+            result = main()
+        assert result == 1
+        captured = capsys.readouterr()
+        assert "ADS_API_KEY" in captured.out
+
+    def test_inspire_source_no_api_key_ok(self, tmp_path, capsys):
+        """Using --source inspire should not require an ADS API key."""
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\cite{Author:2020abc}")
+        with (
+            patch("sys.argv", ["easybib", str(tex), "--source", "inspire"]),
+            patch.dict("os.environ", {}, clear=True),
+            patch("easybib.cli.fetch_bibtex", return_value=(None, None)),
+        ):
+            result = main()
+        # Should not return 1 for missing API key
+        assert result is None
+
+
+class TestAdsApiKeyOverride:
+    def test_flag_overrides_env(self, tmp_path, capsys):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\cite{Author:2020abc}")
+        with (
+            patch(
+                "sys.argv",
+                ["easybib", str(tex), "--ads-api-key", "flag-key"],
+            ),
+            patch.dict("os.environ", {"ADS_API_KEY": "env-key"}, clear=True),
+            patch("easybib.cli.fetch_bibtex") as mock_fetch,
+        ):
+            mock_fetch.return_value = (
+                "@article{Author:2020abc,\n  title={Test},\n  author={Doe, J.},\n}",
+                "ADS",
+            )
+            main()
+        # The flag value should be used, not the env var
+        call_args = mock_fetch.call_args
+        assert call_args[0][1] == "flag-key"
+
+
+class TestFileVsDirectory:
+    def test_single_file(self, tmp_path, capsys):
+        tex = tmp_path / "paper.tex"
+        tex.write_text(r"\cite{A:2020abc}")
+        with patch("sys.argv", ["easybib", str(tex), "--list-keys"]):
+            main()
+        captured = capsys.readouterr()
+        assert "A:2020abc" in captured.out
+
+    def test_directory_recursive(self, tmp_path, capsys):
+        nested = tmp_path / "ch1"
+        nested.mkdir()
+        (nested / "intro.tex").write_text(r"\cite{A:2020abc}")
+        (tmp_path / "main.tex").write_text(r"\cite{B:2021xyz}")
+        with patch("sys.argv", ["easybib", str(tmp_path), "--list-keys"]):
+            main()
+        captured = capsys.readouterr()
+        assert "A:2020abc" in captured.out
+        assert "B:2021xyz" in captured.out

easybib-0.2.0/tests/test_core.py

@@ -0,0 +1,232 @@
+"""Tests for easybib.core pure functions."""
+
+import pytest
+
+from easybib.core import (
+    extract_cite_keys,
+    extract_existing_bib_keys,
+    is_ads_bibcode,
+    is_inspire_key,
+    replace_bibtex_key,
+    truncate_authors,
+)
+
+
+# --- extract_cite_keys ---
+
+
+class TestExtractCiteKeys:
+    def test_basic_cite(self, tmp_path):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\cite{Author:2020abc}")
+        keys, warnings = extract_cite_keys(tex)
+        assert keys == ["Author:2020abc"]
+        assert warnings == []
+
+    def test_citep_and_citet(self, tmp_path):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\citep{A:2020abc} and \citet{B:2021xyz}")
+        keys, warnings = extract_cite_keys(tex)
+        assert set(keys) == {"A:2020abc", "B:2021xyz"}
+        assert warnings == []
+
+    def test_capital_citep(self, tmp_path):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\Citep{Author:2020abc}")
+        keys, warnings = extract_cite_keys(tex)
+        assert keys == ["Author:2020abc"]
+
+    def test_optional_args(self, tmp_path):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\citep[e.g.][]{Author:2020abc}")
+        keys, warnings = extract_cite_keys(tex)
+        assert keys == ["Author:2020abc"]
+
+    def test_multiple_keys_in_single_cite(self, tmp_path):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\cite{A:2020abc, B:2021xyz}")
+        keys, warnings = extract_cite_keys(tex)
+        assert set(keys) == {"A:2020abc", "B:2021xyz"}
+
+    def test_empty_key_warning(self, tmp_path):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\cite{A:2020abc, , B:2021xyz}")
+        keys, warnings = extract_cite_keys(tex)
+        assert set(keys) == {"A:2020abc", "B:2021xyz"}
+        assert len(warnings) == 1
+        assert "Empty citation key" in warnings[0]
+
+    def test_key_without_colon_warning(self, tmp_path):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\cite{nocolon}")
+        keys, warnings = extract_cite_keys(tex)
+        assert keys == []
+        assert len(warnings) == 1
+        assert "not an INSPIRE/ADS key" in warnings[0]
+
+    def test_citeauthor_and_citeyear(self, tmp_path):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\citeauthor{A:2020abc} \citeyear{B:2021xyz}")
+        keys, warnings = extract_cite_keys(tex)
+        assert set(keys) == {"A:2020abc", "B:2021xyz"}
+
+    def test_no_citations(self, tmp_path):
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"No citations here.")
+        keys, warnings = extract_cite_keys(tex)
+        assert keys == []
+        assert warnings == []
+
+    def test_deduplication_not_applied(self, tmp_path):
+        """extract_cite_keys returns all occurrences (dedup is done by the caller)."""
+        tex = tmp_path / "test.tex"
+        tex.write_text(r"\cite{A:2020abc} \cite{A:2020abc}")
+        keys, warnings = extract_cite_keys(tex)
+        assert keys == ["A:2020abc", "A:2020abc"]
+
+
+# --- extract_existing_bib_keys ---
+
+
+class TestExtractExistingBibKeys:
+    def test_parse_keys(self, tmp_path):
+        bib = tmp_path / "refs.bib"
+        bib.write_text(
+            "@article{Author:2020abc,\n"
+            "  title={Test},\n"
+            "}\n\n"
+            "@inproceedings{Other:2021xyz,\n"
+            "  title={Other},\n"
+            "}\n"
+        )
+        keys = extract_existing_bib_keys(bib)
+        assert keys == {"Author:2020abc", "Other:2021xyz"}
+
+    def test_nonexistent_file(self, tmp_path):
+        bib = tmp_path / "missing.bib"
+        keys = extract_existing_bib_keys(bib)
+        assert keys == set()
+
+    def test_empty_file(self, tmp_path):
+        bib = tmp_path / "empty.bib"
+        bib.write_text("")
+        keys = extract_existing_bib_keys(bib)
+        assert keys == set()
+
+
+# --- replace_bibtex_key ---
+
+
+class TestReplaceBibtexKey:
+    def test_simple_replacement(self):
+        bibtex = "@article{OldKey:2020abc,\n  title={Test},\n}"
+        result = replace_bibtex_key(bibtex, "NewKey:2020xyz")
+        assert result.startswith("@article{NewKey:2020xyz,")
+
+    def test_preserves_content(self):
+        bibtex = "@article{OldKey:2020abc,\n  title={Test},\n  author={Doe},\n}"
+        result = replace_bibtex_key(bibtex, "New:2020xyz")
+        assert "title={Test}" in result
+        assert "author={Doe}" in result
+
+    def test_only_replaces_first(self):
+        bibtex = "@article{A:2020abc,\n  note={See also A:2020abc},\n}"
+        result = replace_bibtex_key(bibtex, "B:2020xyz")
+        assert result.startswith("@article{B:2020xyz,")
+        assert "See also A:2020abc" in result
+
+
+# --- truncate_authors ---
+
+
+class TestTruncateAuthors:
+    def test_truncation(self):
+        bibtex = (
+            "@article{Key:2020abc,\n"
+            "  author={Alpha, A. and Beta, B. and Gamma, G. and Delta, D. and Epsilon, E.},\n"
+            "  title={Test},\n"
+            "}"
+        )
+        result = truncate_authors(bibtex, max_authors=3)
+        assert "Alpha, A. and Beta, B. and Gamma, G. and others" in result
+        assert "Delta" not in result
+
+    def test_no_truncation_when_within_limit(self):
+        bibtex = (
+            "@article{Key:2020abc,\n"
+            "  author={Alpha, A. and Beta, B.},\n"
+            "  title={Test},\n"
+            "}"
+        )
+        result = truncate_authors(bibtex, max_authors=3)
+        assert result == bibtex
+
+    def test_max_authors_zero_no_change(self):
+        bibtex = (
+            "@article{Key:2020abc,\n"
+            "  author={Alpha, A. and Beta, B. and Gamma, G. and Delta, D.},\n"
+            "  title={Test},\n"
+            "}"
+        )
+        result = truncate_authors(bibtex, max_authors=0)
+        assert result == bibtex
+
+    def test_max_authors_none_no_change(self):
+        bibtex = (
+            "@article{Key:2020abc,\n"
+            "  author={Alpha, A. and Beta, B. and Gamma, G. and Delta, D.},\n"
+            "  title={Test},\n"
+            "}"
+        )
+        result = truncate_authors(bibtex, max_authors=None)
+        assert result == bibtex
+
+    def test_exact_limit(self):
+        bibtex = (
+            "@article{Key:2020abc,\n"
+            "  author={Alpha, A. and Beta, B. and Gamma, G.},\n"
+            "  title={Test},\n"
+            "}"
+        )
+        result = truncate_authors(bibtex, max_authors=3)
+        assert result == bibtex
+
+
+# --- is_ads_bibcode ---
+
+
+class TestIsAdsBibcode:
+    def test_positive(self):
+        assert is_ads_bibcode("2016PhRvL.116f1102A") is True
+
+    def test_positive_with_ampersand(self):
+        assert is_ads_bibcode("2020A&A...641A...6P") is True
+
+    def test_negative_inspire_key(self):
+        assert is_ads_bibcode("Abbott:2016blz") is False
+
+    def test_negative_short_string(self):
+        assert is_ads_bibcode("2020") is False
+
+    def test_negative_no_leading_year(self):
+        assert is_ads_bibcode("PhRvL.116f1102A") is False
+
+
+# --- is_inspire_key ---
+
+
+class TestIsInspireKey:
+    def test_positive(self):
+        assert is_inspire_key("Abbott:2016blz") is True
+
+    def test_positive_hyphenated(self):
+        assert is_inspire_key("LIGO-Scientific:2020abc") is True
+
+    def test_negative_ads_bibcode(self):
+        assert is_inspire_key("2016PhRvL.116f1102A") is False
+
+    def test_negative_no_colon(self):
+        assert is_inspire_key("Abbott2016blz") is False
+
+    def test_negative_missing_letters(self):
+        assert is_inspire_key("Abbott:2016") is False

easybib-0.2.0/tests/test_fetch.py

@@ -0,0 +1,201 @@
+"""Tests for easybib.core network functions (mocked)."""
+
+from unittest.mock import MagicMock, patch
+
+from easybib.core import (
+    fetch_bibtex,
+    get_ads_bibtex,
+    get_ads_info_from_inspire,
+    get_inspire_bibtex,
+    search_ads_by_arxiv,
+)
+
+SAMPLE_BIBTEX = "@article{Author:2020abc,\n  title={Test},\n  author={Doe, J.},\n}"
+
+
+# --- get_inspire_bibtex ---
+
+
+class TestGetInspireBibtex:
+    @patch("easybib.core.requests.get")
+    def test_success(self, mock_get):
+        mock_get.return_value = MagicMock(status_code=200, text=SAMPLE_BIBTEX)
+        result = get_inspire_bibtex("Author:2020abc")
+        assert result == SAMPLE_BIBTEX.strip()
+
+    @patch("easybib.core.requests.get")
+    def test_empty_response(self, mock_get):
+        mock_get.return_value = MagicMock(status_code=200, text="   ")
+        result = get_inspire_bibtex("Author:2020abc")
+        assert result is None
+
+    @patch("easybib.core.requests.get")
+    def test_non_200(self, mock_get):
+        mock_get.return_value = MagicMock(status_code=404, text="")
+        result = get_inspire_bibtex("Author:2020abc")
+        assert result is None
+
+
+# --- get_ads_bibtex ---
+
+
+class TestGetAdsBibtex:
+    @patch("easybib.core.requests.post")
+    def test_success(self, mock_post):
+        mock_post.return_value = MagicMock(
+            status_code=200,
+            json=MagicMock(return_value={"export": SAMPLE_BIBTEX}),
+        )
+        result = get_ads_bibtex("2020ApJ...000..000A", "fake-key")
+        assert result == SAMPLE_BIBTEX.strip()
+
+    @patch("easybib.core.requests.post")
+    def test_no_records(self, mock_post):
+        mock_post.return_value = MagicMock(
+            status_code=200,
+            json=MagicMock(return_value={"export": "No records found"}),
+        )
+        result = get_ads_bibtex("2020ApJ...000..000A", "fake-key")
+        assert result is None
+
+    @patch("easybib.core.requests.post")
+    def test_non_200(self, mock_post):
+        mock_post.return_value = MagicMock(status_code=500)
+        result = get_ads_bibtex("2020ApJ...000..000A", "fake-key")
+        assert result is None
+
+
+# --- get_ads_info_from_inspire ---
+
+
+class TestGetAdsInfoFromInspire:
+    @patch("easybib.core.requests.get")
+    def test_success(self, mock_get):
+        mock_get.return_value = MagicMock(
+            status_code=200,
+            json=MagicMock(
+                return_value={
+                    "hits": {
+                        "hits": [
+                            {
+                                "metadata": {
+                                    "external_system_identifiers": [
+                                        {"schema": "ADS", "value": "2020ApJ...000..000A"}
+                                    ],
+                                    "arxiv_eprints": [{"value": "2001.12345"}],
+                                }
+                            }
+                        ]
+                    }
+                }
+            ),
+        )
+        bibcode, arxiv_id = get_ads_info_from_inspire("Author:2020abc")
+        assert bibcode == "2020ApJ...000..000A"
+        assert arxiv_id == "2001.12345"
+
+    @patch("easybib.core.requests.get")
+    def test_no_hits(self, mock_get):
+        mock_get.return_value = MagicMock(
+            status_code=200,
+            json=MagicMock(return_value={"hits": {"hits": []}}),
+        )
+        bibcode, arxiv_id = get_ads_info_from_inspire("Author:2020abc")
+        assert bibcode is None
+        assert arxiv_id is None
+
+    @patch("easybib.core.requests.get")
+    def test_non_200(self, mock_get):
+        mock_get.return_value = MagicMock(status_code=500)
+        bibcode, arxiv_id = get_ads_info_from_inspire("Author:2020abc")
+        assert bibcode is None
+        assert arxiv_id is None
+
+
+# --- search_ads_by_arxiv ---
+
+
+class TestSearchAdsByArxiv:
+    @patch("easybib.core.requests.get")
+    def test_success(self, mock_get):
+        mock_get.return_value = MagicMock(
+            status_code=200,
+            json=MagicMock(
+                return_value={
+                    "response": {"docs": [{"bibcode": "2020ApJ...000..000A"}]}
+                }
+            ),
+        )
+        result = search_ads_by_arxiv("2001.12345", "fake-key")
+        assert result == "2020ApJ...000..000A"
+
+    @patch("easybib.core.requests.get")
+    def test_empty_docs(self, mock_get):
+        mock_get.return_value = MagicMock(
+            status_code=200,
+            json=MagicMock(return_value={"response": {"docs": []}}),
+        )
+        result = search_ads_by_arxiv("2001.12345", "fake-key")
+        assert result is None
+
+    @patch("easybib.core.requests.get")
+    def test_non_200(self, mock_get):
+        mock_get.return_value = MagicMock(status_code=403)
+        result = search_ads_by_arxiv("2001.12345", "fake-key")
+        assert result is None
+
+
+# --- fetch_bibtex (source routing) ---
+
+
+class TestFetchBibtex:
+    @patch("easybib.core.get_inspire_bibtex")
+    @patch("easybib.core.get_ads_bibtex")
+    @patch("easybib.core.get_ads_info_from_inspire")
+    def test_ads_source(self, mock_info, mock_ads, mock_inspire):
+        """With source='ads', ADS path is tried (via INSPIRE cross-ref)."""
+        mock_info.return_value = ("2020ApJ...000..000A", None)
+        mock_ads.return_value = SAMPLE_BIBTEX
+        result, source = fetch_bibtex("Author:2020abc", "fake-key", source="ads")
+        assert result == SAMPLE_BIBTEX
+        assert "ADS" in source
+
+    @patch("easybib.core.get_inspire_bibtex")
+    def test_inspire_source(self, mock_inspire):
+        """With source='inspire', INSPIRE is tried first."""
+        mock_inspire.return_value = SAMPLE_BIBTEX
+        result, source = fetch_bibtex("Author:2020abc", "fake-key", source="inspire")
+        assert result == SAMPLE_BIBTEX
+        assert "INSPIRE" in source
+
+    @patch("easybib.core.get_inspire_bibtex")
+    def test_auto_source_inspire_key(self, mock_inspire):
+        """With source='auto' and an INSPIRE-style key, INSPIRE is tried first."""
+        mock_inspire.return_value = SAMPLE_BIBTEX
+        result, source = fetch_bibtex("Author:2020abc", "fake-key", source="auto")
+        assert result == SAMPLE_BIBTEX
+        assert "INSPIRE" in source
+
+    @patch("easybib.core.get_ads_bibtex")
+    def test_auto_source_ads_bibcode(self, mock_ads):
+        """With source='auto' and an ADS-style key, ADS is tried first."""
+        mock_ads.return_value = SAMPLE_BIBTEX
+        result, source = fetch_bibtex(
+            "2016PhRvL.116f1102A", "fake-key", source="auto"
+        )
+        assert result == SAMPLE_BIBTEX
+        assert "ADS" in source
+
+    @patch("easybib.core.get_inspire_bibtex")
+    @patch("easybib.core.get_ads_bibtex")
+    @patch("easybib.core.get_ads_info_from_inspire")
+    @patch("easybib.core.search_ads_by_arxiv")
+    def test_not_found(self, mock_search, mock_info, mock_ads, mock_inspire):
+        """When nothing is found, returns (None, None)."""
+        mock_inspire.return_value = None
+        mock_ads.return_value = None
+        mock_info.return_value = (None, None)
+        mock_search.return_value = None
+        result, source = fetch_bibtex("Author:2020abc", "fake-key", source="ads")
+        assert result is None
+        assert source is None

easybib-0.1.0/src/easybib.egg-info/requires.txt

@@ -1 +0,0 @@
-requests