diff options
Diffstat (limited to 'tests/unit/browser/webkit/test_webkitelem.py')
-rw-r--r-- | tests/unit/browser/webkit/test_webkitelem.py | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/tests/unit/browser/webkit/test_webkitelem.py b/tests/unit/browser/webkit/test_webkitelem.py index 37262a7b3..f12df66f8 100644 --- a/tests/unit/browser/webkit/test_webkitelem.py +++ b/tests/unit/browser/webkit/test_webkitelem.py @@ -19,12 +19,13 @@ """Tests for the webelement utils.""" +from typing import TYPE_CHECKING from unittest import mock import collections.abc import operator import itertools -import attr +import dataclasses import pytest from PyQt5.QtCore import QRect, QPoint, QUrl QWebElement = pytest.importorskip('PyQt5.QtWebKit').QWebElement @@ -33,6 +34,8 @@ from qutebrowser.browser import browsertab from qutebrowser.browser.webkit import webkitelem from qutebrowser.misc import objects from qutebrowser.utils import usertypes +if TYPE_CHECKING: + from helpers import stubs def get_webelem(geometry=None, frame=None, *, null=False, style=None, @@ -527,12 +530,12 @@ class TestIsVisibleIframe: elem1-elem4: FakeWebElements to test. """ - @attr.s + @dataclasses.dataclass class Objects: - frame = attr.ib() - iframe = attr.ib() - elems = attr.ib() + frame: 'stubs.FakeWebFrame' + iframe: 'stubs.FakeWebFrame' + elems: webkitelem.WebKitElement @pytest.fixture def objects(self, stubs): |