1 files changed, 8 insertions, 10 deletions
diff --git a/tests/units/cli/check/test__init__.py b/tests/units/cli/check/test__init__.py
index a3a770b..2501dc8 100644
--- a/tests/units/cli/check/test__init__.py
+++ b/tests/units/cli/check/test__init__.py
@@ -1,30 +1,28 @@
 # Copyright (c) 2023-2024 Arista Networks, Inc.
 # Use of this source code is governed by the Apache License 2.0
 # that can be found in the LICENSE file.
-"""
-Tests for anta.cli.check
-"""
+"""Tests for anta.cli.check."""
+
 from __future__ import annotations
 
-from click.testing import CliRunner
+from typing import TYPE_CHECKING
 
 from anta.cli import anta
 from anta.cli.utils import ExitCode
 
+if TYPE_CHECKING:
+    from click.testing import CliRunner
+
 
 def test_anta_check(click_runner: CliRunner) -> None:
-    """
-    Test anta check
-    """
+    """Test anta check."""
     result = click_runner.invoke(anta, ["check"])
     assert result.exit_code == ExitCode.OK
     assert "Usage: anta check" in result.output
 
 
 def test_anta_check_help(click_runner: CliRunner) -> None:
-    """
-    Test anta check --help
-    """
+    """Test anta check --help."""
     result = click_runner.invoke(anta, ["check", "--help"])
     assert result.exit_code == ExitCode.OK
     assert "Usage: anta check" in result.output