tests

griptape-ai · Oct 21, 2024 · 00435fd · 00435fd
1 parent 4ab3dcf
commit 00435fd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/unit/events/test_action_chunk_event.py b/tests/unit/events/test_action_chunk_event.py
@@ -8,6 +8,7 @@ class TestCompletionChunkEvent:
         {"name": "foo", "tag": None, "path": None, "partial_input": None},
         {"name": "foo", "tag": "bar", "path": None, "partial_input": None},
         {"name": "foo", "tag": "bar", "path": "baz", "partial_input": None},
+        {"name": "foo", "tag": None, "path": "baz", "partial_input": None},
         {"name": "foo", "tag": "bar", "path": "baz", "partial_input": "qux"},
         {"name": None, "tag": None, "path": None, "partial_input": "qux"},
     ]

diff --git a/tests/unit/events/test_text_chunk_event.py b/tests/unit/events/test_text_chunk_event.py
@@ -10,6 +10,7 @@ def text_chunk_event(self):
 
     def test_token(self, text_chunk_event):
         assert text_chunk_event.token == "foo bar"
+        assert str(text_chunk_event) == "foo bar"
 
     def test_to_dict(self, text_chunk_event):
         assert text_chunk_event.to_dict()["token"] == "foo bar"