feat: typing

Author: germa89

tests/conftest.py

@@ -26,6 +26,7 @@
 from pathlib import Path
 from shutil import get_terminal_size
 from sys import platform
+from typing import Literal, Union
 from unittest.mock import patch
 
 from _pytest.terminal import TerminalReporter  # for terminal customization
@@ -124,8 +125,27 @@
     reason="This tests does not work on student version.",
 )
 
+_REQUIRES_ARG = Union[
+    Literal[
+        "GRPC",
+        "DPF",
+        "LOCAL",
+        "REMOTE",
+        "CICD",
+        "NOCICD",
+        "XSERVER",
+        "LINUX",
+        "NOLINUX",
+        "WINDOWS",
+        "NOWINDOWS",
+        "NOSTUDENT",
+        "CONSOLE",
+    ],
+    str,
+]
+
 
-def requires(requirement: str):
+def requires(requirement: _REQUIRES_ARG):
     """Check requirements"""
     requirement = requirement.lower()