Spaces:
Sleeping
Sleeping
File size: 1,952 Bytes
86c402d |
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 47 48 49 50 51 52 53 54 55 |
"""
Конфигурация pytest для тестов ntr_text_fragmentation.
"""
from uuid import UUID
import pytest
from ntr_text_fragmentation.models.linker_entity import LinkerEntity
from tests.custom_entity import CustomEntity # Импортируем наш кастомный класс
@pytest.fixture
def sample_entity():
"""
Фикстура, возвращающая экземпляр LinkerEntity с предустановленными значениями.
"""
return LinkerEntity(
id=UUID('12345678-1234-5678-1234-567812345678'),
name="Тестовая сущность",
text="Текст тестовой сущности",
metadata={"test_key": "test_value"}
)
@pytest.fixture
def sample_custom_entity():
"""
Фикстура, возвращающая экземпляр CustomEntity с предустановленными значениями.
"""
return CustomEntity(
id=UUID('87654321-8765-4321-8765-432187654321'),
name="Тестовый кастомный объект",
text="Текст кастомного объекта",
metadata={"original_key": "original_value"},
in_search_text="Текст для поиска кастомного объекта",
custom_field1="custom_value",
custom_field2=42
)
@pytest.fixture
def sample_link():
"""
Фикстура, возвращающая экземпляр LinkerEntity с предустановленными значениями связи.
"""
return LinkerEntity(
id=UUID('98765432-9876-5432-9876-543298765432'),
name="Тестовая связь",
text="Текст тестовой связи",
metadata={"test_key": "test_value"},
source_id=UUID('12345678-1234-5678-1234-567812345678'),
target_id=UUID('87654321-8765-4321-8765-432187654321'),
type="Link"
) |