tests/test_functions.py

# Tests the functions themselves, not the function calling

from pprint import pprint
from functions import get_analysis

def test_get_analysis():
    query = "How is MSTR doing?"
    output = get_analysis(query)
    
    pprint(output)
    assert len(output) != 0
    assert "MSTR" in output[0]['securities']