github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-01-24 05:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fmt

Browse Source
This commit is contained in:
Bagatur
2024-09-02 15:30:51 -07:00
parent 71661fd16b
commit 8c49af3e9c
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
libs/community/tests/unit_tests/document_loaders/test_recursive_url_loader.py
View File

@@ -122,39 +122,39 @@ def mock_requests(loader):
return docs
def test_sync__init__():
def test_sync__init__() -> None:
loader = RecursiveUrlLoader("http://test.com", max_depth=1)
docs = mock_requests(loader)
assert len(docs) == 1
def test_async__init__(mocker):
def test_async__init__(mocker: Any) -> None:
mocker.patch.object(aiohttp.ClientSession, "get", new=MockGet)
loader = RecursiveUrlLoader("http://test.com", max_depth=1, use_async=True)
docs = loader.load()
assert len(docs) == 1
def test_sync_default_depth():
def test_sync_default_depth() -> None:
loader = RecursiveUrlLoader("http://test.com")
docs = mock_requests(loader)
assert len(docs) == 3
def test_async_default_depth(mocker):
def test_async_default_depth(mocker: Any) -> None:
mocker.patch.object(aiohttp.ClientSession, "get", new=MockGet)
loader = RecursiveUrlLoader("http://test.com", use_async=True)
docs = loader.load()
assert len(docs) == 3
def test_sync_deduplication():
def test_sync_deduplication() -> None:
loader = RecursiveUrlLoader("http://test.com", max_depth=3)
docs = mock_requests(loader)
assert len(docs) == 4
def test_async_deduplication(mocker):
def test_async_deduplication(mocker: Any) -> None:
mocker.patch.object(aiohttp.ClientSession, "get", new=MockGet)
loader = RecursiveUrlLoader("http://test.com", max_depth=3, use_async=True)
docs = loader.load()