From 67271a2e42e9339278b6c1a820dd9d80ae0c1cf1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jiri=20Dan=C4=9Bk?= <jdanek@redhat.com>
Date: Fri, 17 Jan 2025 10:40:05 +0100
Subject: [PATCH] fixup precommit add types

---
 tests/workbenches/docs.py | 36 +++++++++++++++++++++---------------
 1 file changed, 21 insertions(+), 15 deletions(-)

diff --git a/tests/workbenches/docs.py b/tests/workbenches/docs.py
index d84c964..5fcab3e 100644
--- a/tests/workbenches/docs.py
+++ b/tests/workbenches/docs.py
@@ -1,32 +1,38 @@
-def Desc(value: str):
+from typing import Callable
+import typing
+
+T = typing.TypeVar("T")
+
+
+def Desc(value: str) -> str:
     return value
 
 
 def Step(
-    value: str,
-    expected: str,
-):
+        value: str,
+        expected: str,
+) -> tuple[str, str]:
     return value, expected
 
 
 def SuiteDoc(
-    description: str,
-    beforeTestSteps: set[Step],
-    afterTestSteps: set[Step],
-):
+        description: str,
+        beforeTestSteps: set[Step],
+        afterTestSteps: set[Step],
+) -> Callable[[T], T]:
     return lambda x: x
 
 
 def Contact(
-    name: str,
-    email: str,
-):
+        name: str,
+        email: str,
+) -> tuple[str, str]:
     return name, email
 
 
 def TestDoc(
-    description: str,
-    contact: str,
-    steps: set[Step],
-):
+        description: str,
+        contact: str,
+        steps: set[Step],
+) -> Callable[[T], T]:
     return lambda x: x