tooluniverse 1.0.6__py3-none-any.whl → 1.0.7__py3-none-any.whl

tooluniverse/test/test_medlineplus.py

@@ -1,220 +0,0 @@
-# test_medlineplus.py
-
-from tooluniverse import ToolUniverse
-import json
-
-# Step 1: Initialize ToolUniverse and load all tools (including MedlinePlus tools)
-tooluni = ToolUniverse()
-tooluni.load_tools()
-
-# Step 2: Define example inputs
-VALID_TERM_EN = "diabetes"  # English health topic keyword
-VALID_DB_EN = "healthTopics"  # English health topics database
-VALID_TERM_ES = "diabetes"  # Spanish health topic keyword
-VALID_DB_ES = "healthTopicsSpanish"  # Spanish health topics database
-
-# ICD-10 CM examples
-VALID_CS_ICD10 = "2.16.840.1.113883.6.90"
-VALID_CODE_ICD10 = "E11.9"  # Diabetes without complications
-
-# RXCUI examples
-VALID_CS_RXCUI = "2.16.840.1.113883.6.88"
-VALID_CODE_RXCUI = "637188"  # RXCUI for Chantix
-VALID_DN_RXCUI = "Chantix 0.5 MG Oral Tablet"
-
-# Genetics examples
-VALID_CONDITION = "alzheimer-disease"  # Genetic condition URL slug
-VALID_GENE = "BRCA1"  # Gene URL slug
-
-# Invalid examples
-INVALID_TERM = ""  # Empty keyword
-INVALID_CS = "INVALID_CS"  # Invalid code system
-INVALID_CODE = "INVALID_CODE"  # Invalid code value
-
-
-def print_result(result, tool_name):
-    """Print formatted results"""
-    if isinstance(result, dict):
-        if "error" in result:
-            print("⚠️ Error:", result["error"])
-            if "detail" in result:
-                print("Detail:", result["detail"])
-            return
-
-        # Define output formats
-        formats = {
-            "MedlinePlus_search_topics_by_keyword": {
-                "title": "📚 Health Topic Search Results",
-                "items": "topics",
-                "fields": {
-                    "title": "Title",
-                    "meta_desc": "Description",
-                    "url": "Link",
-                    "language": "Language",
-                    "rank": "Rank",
-                    "also_called": "Also Known As",
-                    "summary": "Detailed Summary",
-                    "groups": "Categories",
-                },
-            },
-            "MedlinePlus_connect_lookup_by_code": {
-                "title": "🔍 Code Lookup Results",
-                "items": "responses",
-                "fields": {"title": "Title", "summary": "Summary", "url": "Link"},
-            },
-            "MedlinePlus_get_genetics_condition_by_name": {
-                "title": "🧬 Genetic Condition Information",
-                "fields": {
-                    "name": "Condition Name",
-                    "description": "Description",
-                    "genes": "Related Genes",
-                    "synonyms": "Synonyms",
-                    "ghr_page": "Detailed Information Page",
-                },
-            },
-            "MedlinePlus_get_genetics_gene_by_name": {
-                "title": "🧬 Gene Information",
-                "fields": {
-                    "name": "Gene Name",
-                    "function": "Function Description",
-                    "health_conditions": "Related Health Conditions",
-                    "synonyms": "Synonyms",
-                    "ghr_page": "Detailed Information Page",
-                },
-            },
-            "MedlinePlus_get_genetics_index": {
-                "title": "📋 Genetics Topics List",
-                "items": "topics",
-                "fields": {"name": "Name", "url": "Link"},
-            },
-        }
-
-        # Get tool output format
-        fmt = formats.get(tool_name, {})
-        if not fmt:
-            print("\nRaw Response:")
-            print(json.dumps(result, indent=2, ensure_ascii=False))
-            return
-
-        # Print title
-        print(f"\n{fmt['title']}:")
-
-        # Handle list type responses
-        if "items" in fmt:
-            items = result.get(fmt["items"], [])
-            if not items:
-                print("No related content found")
-                return
-            for i, item in enumerate(items, 1):
-                print(f"\n{i}. {item.get(list(fmt['fields'].keys())[0], '')}")
-                for field, label in fmt["fields"].items():
-                    value = item.get(field, "")
-                    if isinstance(value, list):
-                        print(f"   {label}:")
-                        for v in value:
-                            print(f"   - {v}")
-                    else:
-                        print(f"   {label}: {value}")
-            return
-
-        # Handle dictionary type responses
-        for field, label in fmt["fields"].items():
-            value = result.get(field, "")
-            if isinstance(value, list):
-                print(f"\n{label}:")
-                for item in value:
-                    print(f"- {item}")
-            else:
-                print(f"\n{label}: {value}")
-
-    elif isinstance(result, str):
-        print("\nRaw Response Text:")
-        print(result)
-    else:
-        print("\nOther Type Response:")
-        print(result)
-
-
-# Step 3: Define test case list
-test_queries = [
-    # Health topic search - using simpler search terms
-    {
-        "name": "MedlinePlus_search_topics_by_keyword",
-        "arguments": {"term": "heart", "db": "healthTopics", "rettype": "topic"},
-    },
-    {
-        "name": "MedlinePlus_search_topics_by_keyword",
-        "arguments": {"term": "diabetes", "db": "healthTopics", "rettype": "topic"},
-    },
-    {
-        "name": "MedlinePlus_search_topics_by_keyword",
-        "arguments": {"term": "cancer", "db": "healthTopics", "rettype": "brief"},
-    },
-    {
-        "name": "MedlinePlus_search_topics_by_keyword",
-        "arguments": {
-            "term": "diabetes",
-            "db": "healthTopicsSpanish",
-            "rettype": "topic",
-        },
-    },
-    # Code lookup
-    {
-        "name": "MedlinePlus_connect_lookup_by_code",
-        "arguments": {
-            "cs": "2.16.840.1.113883.6.90",
-            "c": "E11.9",
-            "language": "en",
-            "format": "json",
-        },
-    },
-    {
-        "name": "MedlinePlus_connect_lookup_by_code",
-        "arguments": {
-            "cs": "2.16.840.1.113883.6.88",
-            "c": "637188",
-            "dn": "Chantix 0.5 MG Oral Tablet",
-            "language": "en",
-            "format": "xml",
-        },
-    },
-    # Genetics information
-    {
-        "name": "MedlinePlus_get_genetics_condition_by_name",
-        "arguments": {"condition": "alzheimer-disease", "format": "json"},
-    },
-    {
-        "name": "MedlinePlus_get_genetics_gene_by_name",
-        "arguments": {"gene": "BRCA1", "format": "json"},
-    },
-    {"name": "MedlinePlus_get_genetics_index", "arguments": {}},
-    # Error testing
-    {
-        "name": "MedlinePlus_search_topics_by_keyword",
-        "arguments": {"term": "", "db": "healthTopics", "rettype": "topic"},
-    },
-    {
-        "name": "MedlinePlus_connect_lookup_by_code",
-        "arguments": {
-            "cs": "INVALID_CS",
-            "c": "INVALID_CODE",
-            "language": "en",
-            "format": "json",
-        },
-    },
-]
-
-# Step 4: Loop through all test cases and print example output
-for idx, query in enumerate(test_queries):
-    print(f"\n{'='*80}")
-    print(f"[{idx+1}] Running tool: {query['name']}")
-    print(f"Arguments: {query['arguments']}")
-    print(f"{'='*80}")
-
-    try:
-        result = tooluni.run(query)
-        print_result(result, query["name"])
-    except Exception as e:
-        print("⚠️ Exception:", e)
-
-print("\n🎉 MedlinePlus Testing Complete!")

tooluniverse/test/test_odphp_tool.py

@@ -1,166 +0,0 @@
-import json
-import os
-from tooluniverse import ToolUniverse
-import pytest
-
-schema_path = os.path.join(os.path.dirname(__file__), "..", "data", "odphp_tools.json")
-with open(schema_path) as f:
-    schemas = {tool["name"]: tool["return_schema"] for tool in json.load(f)}
-
-tooluni = ToolUniverse()
-tooluni.load_tools()
-
-
-def summarize_result(tool_name, res):
-    if isinstance(res, str):
-        return f"{tool_name}: INVALID Raw string response: {res[:200]}..."
-    if isinstance(res, dict):
-        if "error" in res:
-            return f"{tool_name}: ERROR {res['error']}"
-        data = res.get("data", {})
-        total = data.get("Total") if isinstance(data, dict) else None
-        msg = f"{tool_name}: SUCCESS"
-        if isinstance(total, int):
-            msg += f" | Total={total}"
-
-        if tool_name == "odphp_myhealthfinder":
-            heading = data.get("MyHFHeading", "")
-            resources = (data.get("Resources", {}).get("All", {}).get("Resource", [])) or []
-            first_title = resources[0].get("Title") if resources else None
-            msg += f" | Heading='{heading[:60]}...'"
-            if first_title:
-                msg += f" | FirstResource='{first_title}'"
-            callouts = (data.get("Callouts", {}).get("All", {}).get("Resource", [])) or []
-            if callouts and callouts[0].get("MyHFTitle"):
-                msg += f" | FirstCallout='{callouts[0].get('MyHFTitle')}'"
-
-        elif tool_name == "odphp_itemlist":
-            items = data.get("Items", {}).get("Item", []) or []
-            titles = [i.get("Title") for i in items[:3]]
-            if titles:
-                msg += f" | ExampleItems={titles}"
-
-        elif tool_name == "odphp_topicsearch":
-            resources = data.get("Resources", {}).get("Resource", []) or []
-            titles = [r.get("Title") for r in resources[:3]]
-            if titles:
-                msg += f" | ExampleTopics={titles}"
-
-        elif tool_name.startswith("odphp_outlink_fetch"):
-            results = res.get("results") or []
-            if results:
-                first = results[0]
-                msg += f" | url={first.get('url')} status={first.get('status')} type={first.get('content_type')}"
-                if first.get("title"):
-                    msg += f" | Title='{first['title'][:50]}...'"
-                if first.get("text"):
-                    snippet = first["text"][:80].replace("\n", " ")
-                    msg += f" | TextSnippet='{snippet}...'"
-
-        expected_keys = schemas.get(tool_name, {}).get("properties", {}).keys()
-        missing = [k for k in expected_keys if k not in data and k not in res]
-        if missing:
-            msg += f" | WARNING: Missing keys {missing}"
-        else:
-            msg += " | Schema OK"
-        return msg
-    return f"{tool_name}: INVALID Unexpected type {type(res)}"
-
-
-def test_01_myhealthfinder_valid():
-    res = tooluni.run({"name": "odphp_myhealthfinder",
-                       "arguments": {"age": 35, "sex": "female", "pregnant": "no", "lang": "en"}})
-    print(summarize_result("odphp_myhealthfinder", res))
-    assert isinstance(res, dict) and not res.get("error")
-
-
-def test_02_itemlist_valid():
-    res = tooluni.run({"name": "odphp_itemlist", "arguments": {"type": "topic", "lang": "en"}})
-    print(summarize_result("odphp_itemlist", res))
-    assert isinstance(res, dict) and not res.get("error")
-
-
-def test_03_topicsearch_keyword_valid():
-    res = tooluni.run({"name": "odphp_topicsearch", "arguments": {"keyword": "cancer", "lang": "en"}})
-    print(summarize_result("odphp_topicsearch", res))
-    assert isinstance(res, dict) and not res.get("error")
-
-
-def test_04_invalid_types_fail_fast():
-    r1 = tooluni.run({"name": "odphp_myhealthfinder", "arguments": {"age": "banana"}})
-    r2 = tooluni.run({"name": "odphp_topicsearch", "arguments": {"topicId": 123}})
-    print("Expected type errors:", r1, r2)
-    assert isinstance(r1, str) and "Type mismatches" in r1
-    assert isinstance(r2, str) and "Type mismatches" in r2
-
-
-def test_05_sections_case_and_strip_html():
-    res = tooluni.run({"name": "odphp_topicsearch",
-                       "arguments": {"keyword": "Keep Your Heart Healthy", "lang": "en", "strip_html": True}})
-    print(summarize_result("odphp_topicsearch", res))
-    assert isinstance(res, dict) and not res.get("error")
-    data = res.get("data") or {}
-    resources = (data.get("Resources") or {}).get("Resource") or []
-    if resources:
-        s_any = resources[0].get("Sections", {})
-        arr = s_any.get("Section") or s_any.get("section") or []
-        assert isinstance(arr, list)
-        assert "PlainSections" in resources[0]
-
-
-def test_06_outlink_fetch_accessible_version():
-    url = "https://odphp.health.gov/myhealthfinder/health-conditions/heart-health/keep-your-heart-healthy"
-    res = tooluni.run({"name": "odphp_outlink_fetch",
-                       "arguments": {"urls": [url], "max_chars": 4000}})
-    print(summarize_result("odphp_outlink_fetch", res))
-    assert isinstance(res, dict) and not res.get("error")
-    results = res.get("results") or []
-    assert results and results[0].get("status") in (200, 301, 302)
-    if "text/html" in (results[0].get("content_type") or ""):
-        assert len(results[0].get("text", "")) > 100
-
-
-def test_07_itemlist_spanish():
-    res = tooluni.run({"name": "odphp_itemlist", "arguments": {"type": "topic", "lang": "es"}})
-    print(summarize_result("odphp_itemlist", res))
-    assert isinstance(res, dict) and not res.get("error")
-
-
-def test_08_topicsearch_by_category():
-    cats = tooluni.run({"name": "odphp_itemlist", "arguments": {"type": "category", "lang": "en"}})
-    first_cat = (cats.get("data", {}).get("Items", {}).get("Item") or [])[0]
-    cid = first_cat.get("Id")
-    res = tooluni.run({"name": "odphp_topicsearch", "arguments": {"categoryId": cid, "lang": "en"}})
-    print(summarize_result("odphp_topicsearch", res))
-    assert isinstance(res, dict) and not res.get("error")
-
-def test_09_outlink_fetch_pdf():
-    url = "https://odphp.health.gov/sites/default/files/2021-12/DGA_Pregnancy_FactSheet-508.pdf"
-    res = tooluni.run({"name": "odphp_outlink_fetch",
-                       "arguments": {"urls": [url], "max_chars": 1000}})
-    print(summarize_result("odphp_outlink_fetch_pdf", res))
-
-    assert isinstance(res, dict) and not res.get("error")
-    results = res.get("results") or []
-    assert results, "No results returned for PDF URL"
-
-    ctype = results[0].get("content_type", "")
-    assert ctype.startswith("application/pdf"), f"Expected PDF but got {ctype}"
-
-    # Ensure text extraction worked at least partially
-    text = results[0].get("text", "")
-    assert text and len(text) > 50, "Extracted PDF text too short"
-
-
-if __name__ == "__main__":
-    print("\nRunning ODPHP tool tests...\n")
-    test_01_myhealthfinder_valid()
-    test_02_itemlist_valid()
-    test_03_topicsearch_keyword_valid()
-    test_04_invalid_types_fail_fast()
-    test_05_sections_case_and_strip_html()
-    test_06_outlink_fetch_accessible_version()
-    test_07_itemlist_spanish()
-    test_08_topicsearch_by_category()
-    test_09_outlink_fetch_pdf()
-    print("\nAll ODPHP tests executed.\n")

tooluniverse/test/test_openalex_tool.py

@@ -1,32 +0,0 @@
-# === openfda_test_all_tools.py ===
-
-from tooluniverse import ToolUniverse
-
-# Step 1: Initialize tool universe
-tooluni = ToolUniverse()
-tooluni.load_tools()
-
-# Step 2: Define test queries for the openalex tool
-test_queries = [
-    {
-        "name": "openalex_literature_search",
-        "arguments": {
-            "search_keywords": "cancer",
-            "max_results": 10,
-            "year_from": 2020,
-            "year_to": 2023,
-            "open_access": True,
-        },
-    },
-]
-
-# Step 3: Run all test queries
-for idx, query in enumerate(test_queries):
-    print(
-        f"\n[{idx+1}] Running tool: {query['name']} with arguments: {query['arguments']}"
-    )
-    result = tooluni.run(query)
-    print("✅ Success. Example output snippet:")
-    print(
-        result if isinstance(result, dict) else str(result)
-    )  # Print snippet if result is big