mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
2f087d63af
Filter out kwargs from inferred schema when determining if a tool is single input. Add a couple unit tests. Move tool unit tests to the tools dir
24 lines
688 B
Python
24 lines
688 B
Python
"""Test Python REPL Tools."""
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
from langchain.tools.python.tool import PythonAstREPLTool, PythonREPLTool
|
|
|
|
|
|
def test_python_repl_tool_single_input() -> None:
|
|
"""Test that the python REPL tool works with a single input."""
|
|
tool = PythonREPLTool()
|
|
assert tool.is_single_input
|
|
assert int(tool.run("print(1 + 1)").strip()) == 2
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
sys.version_info < (3, 9), reason="Requires python version >= 3.9 to run."
|
|
)
|
|
def test_python_ast_repl_tool_single_input() -> None:
|
|
"""Test that the python REPL tool works with a single input."""
|
|
tool = PythonAstREPLTool()
|
|
assert tool.is_single_input
|
|
assert tool.run("1 + 1") == 2
|