From 208208132c4198b8ae4221e237fcfd6b4ac815c5 Mon Sep 17 00:00:00 2001 From: Marcel Wilson Date: Thu, 1 Feb 2024 11:14:11 -0600 Subject: [PATCH] linter fixes --- setup_selenium/selenium_module.py | 3 ++- tests/test_setup.py | 5 ++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/setup_selenium/selenium_module.py b/setup_selenium/selenium_module.py index 2df57b7..1f91573 100644 --- a/setup_selenium/selenium_module.py +++ b/setup_selenium/selenium_module.py @@ -1,11 +1,12 @@ """Setup selenium for testing""" + from __future__ import annotations import errno import logging import os as os from enum import Enum -from typing import TYPE_CHECKING, Union, Optional +from typing import TYPE_CHECKING, Optional, Union from selenium import webdriver from selenium.common.exceptions import NoSuchWindowException, WebDriverException diff --git a/tests/test_setup.py b/tests/test_setup.py index d7d3f5c..36ceb95 100644 --- a/tests/test_setup.py +++ b/tests/test_setup.py @@ -203,18 +203,21 @@ def test_create_firefox_wrong_options() -> None: Browser.FIREFOX, options=SetupSelenium.chrome_options() ) + def test_create_chrome_wrong_options() -> None: with pytest.raises(AssertionError): SetupSelenium.create_driver( Browser.CHROME, options=SetupSelenium.edge_options() ) - + + def test_create_edge_wrong_options() -> None: with pytest.raises(AssertionError): SetupSelenium.create_driver( Browser.EDGE, options=SetupSelenium.firefox_options() ) + @pytest.fixture() def create_logger() -> logging.Logger: """Create a logger."""