griptape

1
from tests.utils.structure_tester import StructureTester
2
import pytest
3

4

5
class TestPromptTask:
6
    @pytest.fixture(
7
        autouse=True, params=StructureTester.PROMPT_TASK_CAPABLE_PROMPT_DRIVERS, ids=StructureTester.prompt_driver_id_fn
8
    )
9
    def structure_tester(self, request):
10
        from griptape.structures import Agent
11

12
        return StructureTester(Agent(conversation_memory=None, prompt_driver=request.param))
13

14
    def test_prompt_task(self, structure_tester):
15
        structure_tester.run("Write a poem about pirates.")
16