diff --git a/tests/great_ai/test_basic_starters.py b/tests/great_ai/test_basic_starters.py index c5c77b2..817d28e 100644 --- a/tests/great_ai/test_basic_starters.py +++ b/tests/great_ai/test_basic_starters.py @@ -19,6 +19,12 @@ class TestHumanReadableToByte(unittest.TestCase): assert hello_world("andras").output == "Hello andras!" + @GreatAI.create + def hello_world(name): + return f"Hello {name}!" + + assert hello_world("andras").output == "Hello andras!" + @GreatAI.create() def hello_world(name: str) -> str: return f"Hello {name}!"