Add type hints to test file for improved clarity and type checking

hyperskill · Feb 14, 2025 · faac723 · faac723
1 parent 93ec52f
commit faac723
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/tests/projects/go/coffee_machine/stage4/tests.py b/tests/projects/go/coffee_machine/stage4/tests.py
@@ -1,5 +1,8 @@
+from typing import List
+
 from hstest.stage_test import *
 from hstest.test_case import TestCase
+from hstest.check_result import CheckResult
 
 CheckResult.correct = lambda: CheckResult(True, '')
 CheckResult.wrong = lambda feedback: CheckResult(False, feedback)

diff --git a/tests/projects/go/coffee_machine/stage5/tests.py b/tests/projects/go/coffee_machine/stage5/tests.py
@@ -1,5 +1,8 @@
+from typing import List
+
 from hstest.stage_test import *
 from hstest.test_case import TestCase
+from hstest.check_result import CheckResult
 
 CheckResult.correct = lambda: CheckResult(True, '')
 CheckResult.wrong = lambda feedback: CheckResult(False, feedback)