summaryrefslogtreecommitdiff
path: root/tests/test_templates.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_templates.py')
-rw-r--r--tests/test_templates.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/tests/test_templates.py b/tests/test_templates.py
index 5fcf893..d61f98a 100644
--- a/tests/test_templates.py
+++ b/tests/test_templates.py
@@ -2,9 +2,13 @@
import pytest
-from mml_composemusic_mcp.server import compose_mml
+from mml_composemusic_mcp.server import _execute_compose_mml
from mml_composemusic_mcp.templates import DESCRIPTIONS, TEMPLATES, get_template
+
+def compose_mml(**kwargs):
+ return _execute_compose_mml(**kwargs).result
+
TEMPLATE_NAMES = (
"basic",
"melody",