Вы можете добавить подсказку типа для атрибута driver
класса TestDemo
:
from typing import ClassVar
from selenium.webdriver import Chrome
class TestDemo:
driver: ClassVar[Chrome]
def test_e2e(self, setup):
...
Конечно, это не будет совместимо с проверкой типов stati c, например mypy
(если вы используете его для набора тестов), поскольку библиотека selenium
не типизирована сама по себе, но ее должно быть достаточно, чтобы IDE могли подобрать предложения кода.
Изменить:
Введение подсказок типа ClassVar
работает с Visual Studio Code и Wing IDE, но, к сожалению, еще не с PyCharm, поскольку в нем отсутствует полная поддержка PEP 526, см. PY-20811 ). Чтобы получить завершение кода с помощью PyCharm, используйте подсказки типа для атрибутов экземпляра:
from typing import ClassVar
from selenium.webdriver import Chrome
class TestDemo:
driver: Chrome
def test_e2e(self, setup):
...