griptape

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

4

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

13
        agent = Agent(prompt_driver=request.param, rules=[Rule("Your name is Tony.")])
14

15
        return StructureTester(agent)
16

17
    def test_rule(self, structure_tester):
18
        structure_tester.run("What is your name?")
19