forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
2 years ago
|
"""Test input manipulating logic."""
|
||
|
|
||
|
import sys
|
||
|
from io import StringIO
|
||
|
|
||
|
from langchain.input import ChainedInput
|
||
|
|
||
|
|
||
|
def test_chained_input_not_verbose() -> None:
|
||
|
"""Test chained input logic."""
|
||
|
old_stdout = sys.stdout
|
||
|
sys.stdout = mystdout = StringIO()
|
||
|
chained_input = ChainedInput("foo")
|
||
|
sys.stdout = old_stdout
|
||
|
output = mystdout.getvalue()
|
||
|
assert output == ""
|
||
|
assert chained_input.input == "foo"
|
||
|
|
||
|
old_stdout = sys.stdout
|
||
|
sys.stdout = mystdout = StringIO()
|
||
|
chained_input.add("bar")
|
||
|
sys.stdout = old_stdout
|
||
|
output = mystdout.getvalue()
|
||
|
assert output == ""
|
||
|
assert chained_input.input == "foobar"
|
||
|
|
||
|
|
||
|
def test_chained_input_verbose() -> None:
|
||
|
"""Test chained input logic, making sure verbose doesn't mess it up."""
|
||
|
old_stdout = sys.stdout
|
||
|
sys.stdout = mystdout = StringIO()
|
||
|
chained_input = ChainedInput("foo", verbose=True)
|
||
|
sys.stdout = old_stdout
|
||
|
output = mystdout.getvalue()
|
||
|
assert output == "foo"
|
||
|
assert chained_input.input == "foo"
|
||
|
|
||
|
old_stdout = sys.stdout
|
||
|
sys.stdout = mystdout = StringIO()
|
||
|
chained_input.add("bar")
|
||
|
sys.stdout = old_stdout
|
||
|
output = mystdout.getvalue()
|
||
|
assert output == "bar"
|
||
|
assert chained_input.input == "foobar"
|
||
|
|
||
|
old_stdout = sys.stdout
|
||
|
sys.stdout = mystdout = StringIO()
|
||
|
chained_input.add("baz", color="blue")
|
||
|
sys.stdout = old_stdout
|
||
|
output = mystdout.getvalue()
|
||
|
assert output == "\x1b[104mbaz\x1b[0m"
|
||
|
assert chained_input.input == "foobarbaz"
|