Add basic tests

tests/great_ai/test_basic_starters.py

@@ -0,0 +1,59 @@
+import unittest
+from functools import lru_cache
+
+import pytest
+
+from src.great_ai import (
+    ArgumentValidationError,
+    GreatAI,
+    WrongDecoratorOrderError,
+    parameter,
+)
+
+
+class TestHumanReadableToByte(unittest.TestCase):
+    def test_create_simple_cases(self) -> None:
+        @GreatAI.create
+        def hello_world(name: str) -> str:
+            return f"Hello {name}!"
+
+        assert hello_world("andras").output == "Hello andras!"
+
+        @GreatAI.create()
+        def hello_world(name: str) -> str:
+            return f"Hello {name}!"
+
+        assert hello_world("andras").output == "Hello andras!"
+
+    def test_create_with_other_decorator(self) -> None:
+        @GreatAI.create
+        @lru_cache
+        def hello_world(name: str) -> str:
+            return f"Hello {name}!"
+
+        assert hello_world("andras").output == "Hello andras!"
+
+        @lru_cache
+        @GreatAI.create()
+        def hello_world(name: str) -> str:
+            return f"Hello {name}!"
+
+        assert hello_world("andras").output == "Hello andras!"
+
+    def test_with_parameter(self) -> None:
+        @GreatAI.create
+        @parameter("name", validator=lambda v: len(v) > 5)
+        def hello_world(name: str) -> str:
+            return f"Hello {name}!"
+
+        assert hello_world("andras").output == "Hello andras!"
+
+        with pytest.raises(ArgumentValidationError):
+            hello_world("short")
+
+        with pytest.raises(WrongDecoratorOrderError):
+
+            @parameter("name", validator=lambda v: len(v) > 5)
+            @GreatAI.create
+            def hello_world(name: str) -> str:
+                return f"Hello {name}!"