highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-03-26 17:51:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix mypy for mediawiki tests

Browse Source
This commit is contained in:
Weves 2024-05-25 17:15:03 -07:00 committed by Chris Weaver
parent 26ef5b897d
commit 97d058b8b2
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/tests/unit/danswer/connectors/mediawiki/test_mediawiki_family.py
View File

@ -2,8 +2,8 @@ from typing import Final
from unittest import mock
import pytest
from pywikibot.families.wikipedia_family import Family as WikipediaFamily
from pywikibot.family import Family
from pywikibot.families.wikipedia_family import Family as WikipediaFamily # type: ignore[import-untyped]
from pywikibot.family import Family # type: ignore[import-untyped]
from danswer.connectors.mediawiki import family

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

@ -4,7 +4,7 @@ import datetime
from collections.abc import Iterable
import pytest
import pywikibot
import pywikibot # type: ignore[import-untyped]
from pytest_mock import MockFixture
from danswer.connectors.mediawiki import wiki
@ -95,7 +95,7 @@ def test_get_doc_from_page(site: pywikibot.Site) -> None:
assert (
section.text.strip() == expected_section.strip()
) # Extra whitespace before/after is okay
assert section.link.startswith(test_page.full_url())
assert section.link and section.link.startswith(test_page.full_url())
assert doc.semantic_identifier == test_page.title()
assert doc.metadata == {
"categories": [category.title() for category in test_page.categories()]