def test_prompt_loading()

in utils/prompt_loader.py [0:0]


def test_prompt_loading():
    loader = PromptLoader()
    
    # TEST MySQL
    mysql_context = {
        'meta_data': 'mock_metadata',
        'query': 'mock_query',
        'db_type': 'hologres'  
    }
    mysql_prompt = loader.get_prompt('mysql', mysql_context)
    print("MySQL Prompt Output:\n", mysql_prompt)
    assert "LIMIT n" in mysql_prompt