highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-27 20:38:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix removed parameter in MediaWikiConnector (#1970)

Browse Source
This commit is contained in:
rkuo-danswer
2024-07-29 11:47:30 -07:00
committed by GitHub
parent ea9a9cb553
commit 4a0a927a64
1 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/tests/unit/danswer/connectors/mediawiki/test_wiki.py
View File

@@ -115,27 +115,20 @@ def test_mediawiki_connector_recurse_depth() -> None:
hostname = "wikipedia.org"
categories: list[str] = []
pages = ["Test Page"]
connector_name = "Test Connector"
# Recurse depth less than -1 raises ValueError
with pytest.raises(ValueError):
recurse_depth = -2
wiki.MediaWikiConnector(
hostname, categories, pages, recurse_depth, connector_name
)
wiki.MediaWikiConnector(hostname, categories, pages, recurse_depth)
# Recurse depth of -1 gets parsed as `True`
recurse_depth = -1
connector = wiki.MediaWikiConnector(
hostname, categories, pages, recurse_depth, connector_name
)
connector = wiki.MediaWikiConnector(hostname, categories, pages, recurse_depth)
assert connector.recurse_depth is True
# Recurse depth of 0 or greater gets parsed as an integer
recurse_depth = 0
connector = wiki.MediaWikiConnector(
hostname, categories, pages, recurse_depth, connector_name
)
connector = wiki.MediaWikiConnector(hostname, categories, pages, recurse_depth)
assert connector.recurse_depth == recurse_depth