more tests

poking-agents · Jan 20, 2025 · 4c95f66 · 4c95f66
1 parent 5fee6db
commit 4c95f66
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/modules/test_tools.py b/tests/modules/test_tools.py
@@ -292,6 +292,7 @@ async def test_return_fn(submission: Any, expected_args: str, mocker: MockerFixt
         ("", "", [], False),
         (123, "123", [], False),
         (123, "123", ['123'], True),
+        (123, "123", ['456', '123', '789'], True),
         (123, "123", ['12', '3', '456', '7'], False),
         (["a", "b"], '["a", "b"]', [], False),
         (["a", "b"], '["a", "b"]', ['["a", "b"]'], True),
@@ -300,6 +301,7 @@ async def test_return_fn(submission: Any, expected_args: str, mocker: MockerFixt
         (True, "true", ['true'], True),
         ({"key": "value"}, '{"key": "value"}', [], False),
         ({"key": "value"}, '{"key": "value"}', ['{"key": "value"}'], True),
+        ({"key": "value"}, '{"key": "value"}', ['{"key": "not-value"}','{"key": "value"}'], True),
         ("repeat submission", "repeat submission", ["repeat submission"], True),
         (None, "", [""], True),
     ],