1 files changed, 67 insertions, 0 deletions
diff --git a/tests/test_network_hybrid.py b/tests/test_network_hybrid.py
new file mode 100644
index 0000000..d2912d4
--- /dev/null
+++ b/tests/test_network_hybrid.py
@@ -0,0 +1,67 @@
+"""Test hybrid search: local xfetch with NCBI efetch fallback.
+
+Requires EDIRECT_LOCAL_ARCHIVE and internet access.
+
+Run with: EDIRECT_LOCAL_ARCHIVE=/export3/PubMed/Source python3 -m unittest tests.test_network_hybrid -v
+"""
+
+import os
+import subprocess
+import unittest
+
+ARCHIVE = os.environ.get("EDIRECT_LOCAL_ARCHIVE", "/export3/PubMed/Source")
+
+@unittest.skipUnless(os.path.isdir(ARCHIVE),
+                     f"EDIRECT_LOCAL_ARCHIVE not found: {ARCHIVE}")
+class TestNetworkHybrid(unittest.TestCase):
+    def test_hybrid_matches_esearch(self):
+        """Hybrid xfetch+efetch should return same PMIDs as pure esearch."""
+        env = os.environ.copy()
+        env["EDIRECT_LOCAL_ARCHIVE"] = ARCHIVE
+        query = "(stress) AND (Penk [tiab])"
+        extract = "xtract -pattern PubmedArticle -element MedlineCitation/PMID"
+
+        # Step 1: get PMIDs from NCBI
+        r1 = subprocess.run(
+            ["sh", "-c",
+             f'esearch -db pubmed -query "{query}" | efetch -format uid'],
+            capture_output=True, text=True, timeout=120, env=env)
+        self.assertEqual(r1.returncode, 0, r1.stderr)
+        ncbi_pmids = sorted(set(r1.stdout.strip().split("\n")))
+        ncbi_pmids = [p for p in ncbi_pmids if p.strip()]
+        print(f"  NCBI esearch: {len(ncbi_pmids)} PMIDs")
+
+        # Step 2: local xfetch
+        pmid_str = "\\n".join(ncbi_pmids)
+        r2 = subprocess.run(
+            ["sh", "-c",
+             f'printf "{pmid_str}" | xfetch -db pubmed | {extract}'],
+            capture_output=True, text=True, timeout=120, env=env)
+        local_pmids = set(r2.stdout.strip().split("\n")) - {""}
+        print(f"  Local xfetch: {len(local_pmids)} abstracts")
+
+        # Step 3: fallback efetch for missing
+        missing = [p for p in ncbi_pmids if p not in local_pmids]
+        print(f"  Missing from local: {len(missing)}")
+        fallback_pmids = set()
+        if missing:
+            missing_str = "\\n".join(missing)
+            r3 = subprocess.run(
+                ["sh", "-c",
+                 f'printf "{missing_str}" | efetch -db pubmed -format xml | {extract}'],
+                capture_output=True, text=True, timeout=120, env=env)
+            fallback_pmids = set(r3.stdout.strip().split("\n")) - {""}
+            print(f"  NCBI fallback: {len(fallback_pmids)} abstracts")
+
+        hybrid_pmids = sorted(local_pmids | fallback_pmids)
+        print(f"  Hybrid total: {len(hybrid_pmids)} abstracts")
+
+        # Some articles have no abstract (letters, editorials) so
+        # hybrid may be slightly less than NCBI. Allow up to 5% gap.
+        gap = len(ncbi_pmids) - len(hybrid_pmids)
+        print(f"  Gap: {gap} PMIDs without abstracts")
+        self.assertLessEqual(gap, max(1, len(ncbi_pmids) // 20),
+                             f"Too many missing: hybrid {len(hybrid_pmids)} vs NCBI {len(ncbi_pmids)}")
+
+if __name__ == "__main__":
+    unittest.main()