mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-23 20:23:00 +00:00
zeekay
ce32103320
- Remove all unconditional pytest.skip() calls - Fix file content tests to handle both strict/non-strict modes - Fix missing device_capabilities module in hanzo-network - Fix HanzoNetProvider imports and exports - Fix local_llm.py import errors - Create test helper classes for pagination tests - Fix FastMCP paginator test constructor args - Configure all API keys from ~/.zsh/after.zsh - All tests now pass or skip conditionally based on actual requirements
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
from pathlib import Path
|
|
|
|
import anyio
|
|
import pytest
|
|
from dirty_equals import IsDict, IsList, IsBytes, IsTuple
|
|
|
|
from hanzoai._files import to_httpx_files, async_to_httpx_files
|
|
|
|
readme_path = Path(__file__).parent.parent.joinpath("README.md")
|
|
|
|
|
|
def test_pathlib_includes_file_name() -> None:
|
|
result = to_httpx_files({"file": readme_path})
|
|
print(result)
|
|
assert result == IsDict({"file": IsTuple("README.md", IsBytes())})
|
|
|
|
|
|
def test_tuple_input() -> None:
|
|
result = to_httpx_files([("file", readme_path)])
|
|
print(result)
|
|
assert result == IsList(IsTuple("file", IsTuple("README.md", IsBytes())))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_pathlib_includes_file_name() -> None:
|
|
result = await async_to_httpx_files({"file": readme_path})
|
|
print(result)
|
|
assert result == IsDict({"file": IsTuple("README.md", IsBytes())})
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_supports_anyio_path() -> None:
|
|
result = await async_to_httpx_files({"file": anyio.Path(readme_path)})
|
|
print(result)
|
|
assert result == IsDict({"file": IsTuple("README.md", IsBytes())})
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_tuple_input() -> None:
|
|
result = await async_to_httpx_files([("file", readme_path)])
|
|
print(result)
|
|
assert result == IsList(IsTuple("file", IsTuple("README.md", IsBytes())))
|
|
|
|
|
|
def test_string_not_allowed() -> None:
|
|
with pytest.raises(
|
|
TypeError,
|
|
match="Expected file types input to be a FileContent type or to be a tuple",
|
|
):
|
|
to_httpx_files(
|
|
{
|
|
"file": "foo", # type: ignore
|
|
}
|
|
)
|