mirror of
https://github.com/remsky/Kokoro-FastAPI.git
synced 2025-04-13 09:39:17 +00:00
4b521f9bf0
- Refactored TTS model initialization methods in tts_gpu.py and tts_cpu.py - Added custom logger configuration in main.py - Deprecated text_processing router -> development route
44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
"""Tests for text chunking service"""
|
|
|
|
import pytest
|
|
from unittest.mock import patch
|
|
from api.src.services.text_processing import chunker
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def mock_settings():
|
|
"""Mock settings for all tests"""
|
|
with patch('api.src.services.text_processing.chunker.settings') as mock_settings:
|
|
mock_settings.max_chunk_size = 300
|
|
yield mock_settings
|
|
|
|
|
|
def test_split_text():
|
|
"""Test text splitting into sentences"""
|
|
text = "First sentence. Second sentence! Third sentence?"
|
|
sentences = list(chunker.split_text(text))
|
|
assert len(sentences) == 3
|
|
assert sentences[0] == "First sentence."
|
|
assert sentences[1] == "Second sentence!"
|
|
assert sentences[2] == "Third sentence?"
|
|
|
|
|
|
def test_split_text_empty():
|
|
"""Test splitting empty text"""
|
|
assert list(chunker.split_text("")) == []
|
|
|
|
|
|
def test_split_text_single_sentence():
|
|
"""Test splitting single sentence"""
|
|
text = "Just one sentence."
|
|
assert list(chunker.split_text(text)) == ["Just one sentence."]
|
|
|
|
|
|
def test_split_text_with_custom_chunk_size():
|
|
"""Test splitting with custom max chunk size"""
|
|
text = "First part, second part, third part."
|
|
chunks = list(chunker.split_text(text, max_chunk=15))
|
|
assert len(chunks) == 3
|
|
assert chunks[0] == "First part,"
|
|
assert chunks[1] == "second part,"
|
|
assert chunks[2] == "third part."
|