test_sample.py

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():
    # Test case 1
    input_chat_log = ['hello', 'world\n', 'how', 'are', 'you\n']
    expected_output = '  how are you '
    assert clean_chat_log(input_chat_log) == expected_output

    # Test case 2
    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)