|
import pytest |
|
from streamlit_app import ask, clean_chat_log |
|
|
|
def test_ask(): |
|
question = "What's your name?" |
|
chat_log = "" |
|
model = 'text-davinci-003' |
|
temp = 0.9 |
|
|
|
answer, log = ask(question, chat_log, model, temp) |
|
|
|
assert isinstance(answer, str) |
|
assert isinstance(log, str) |
|
|
|
def test_clean_chat_log(): |
|
|
|
input_chat_log = ['hello', 'world\n', 'how', 'are', 'you\n'] |
|
expected_output = ' how are you ' |
|
assert clean_chat_log(input_chat_log) == expected_output |
|
|
|
|
|
input_chat_log = ['\n', 'this', 'is', 'a', 'new', 'line\n', '\n'] |
|
expected_output = ' this is a new line ' |
|
output = clean_chat_log(input_chat_log) |
|
assert output == expected_output |
|
assert isinstance(output, str) |
|
|
|
|
