Add test suite for all modules
This commit is contained in:
57
local_code_assistant/tests/conftest.py
Normal file
57
local_code_assistant/tests/conftest.py
Normal file
@@ -0,0 +1,57 @@
|
||||
"""Configuration for Local Code Assistant tests."""
|
||||
|
||||
import pytest
|
||||
from click.testing import CliRunner
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def cli_runner():
|
||||
"""Create a CLI runner for testing."""
|
||||
return CliRunner()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mock_config():
|
||||
"""Create a mock configuration."""
|
||||
class MockConfig:
|
||||
ollama_base_url = "http://localhost:11434"
|
||||
ollama_model = "codellama"
|
||||
ollama_timeout = 8000
|
||||
streaming = True
|
||||
default_language = "python"
|
||||
temperature = 0.2
|
||||
max_tokens = 4000
|
||||
syntax_highlighting = True
|
||||
clipboard_enabled = True
|
||||
|
||||
return MockConfig()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def sample_code():
|
||||
"""Provide sample code for testing."""
|
||||
return '''
|
||||
def calculate_fibonacci(n: int) -> list[int]:
|
||||
"""Calculate Fibonacci sequence up to n.
|
||||
|
||||
Args:
|
||||
n: The number of Fibonacci numbers to generate.
|
||||
|
||||
Returns:
|
||||
List of Fibonacci numbers.
|
||||
"""
|
||||
if n <= 0:
|
||||
return []
|
||||
if n == 1:
|
||||
return [0]
|
||||
|
||||
fib = [0, 1]
|
||||
for i in range(2, n):
|
||||
fib.append(fib[i-1] + fib[i-2])
|
||||
|
||||
return fib
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
print(calculate_fibonacci(10))
|
||||
'''
|
||||
Reference in New Issue
Block a user