File size: 1,399 Bytes
2182a08 60532a1 2182a08 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
import pytest
import tempfile
import git
from pathlib import Path
from typing import Dict
from knowlang.configs.config import AppConfig, ParserConfig, LanguageConfig
from knowlang.parser.languages.python.parser import PythonParser
from tests.test_data.python_files import TEST_FILES
@pytest.fixture
def test_config() -> AppConfig:
"""Provides test configuration"""
return AppConfig(
parser=ParserConfig(
languages={
"python": LanguageConfig(
file_extensions=[".py"],
tree_sitter_language="python",
max_file_size=1_000_000,
chunk_types=["class_definition", "function_definition"]
)
}
)
)
@pytest.fixture
def temp_repo():
"""Create a temporary git repository with sample Python files"""
with tempfile.TemporaryDirectory() as temp_dir:
repo = git.Repo.init(temp_dir)
for filename, content in TEST_FILES.items():
file_path = Path(temp_dir) / filename
file_path.write_text(content)
repo.index.add([str(file_path)])
repo.index.commit("Initial commit")
yield temp_dir
@pytest.fixture
def python_parser(test_config):
"""Provides initialized Python parser"""
parser = PythonParser(test_config)
parser.setup()
return parser |