Слишком переусложненная структура - но оставим для будущего

tests/__init__.py

@@ -0,0 +1 @@
+"""Tests package."""

tests/integration/__init__.py

@@ -0,0 +1 @@
+"""Integration tests package."""

tests/test_models.py

@@ -0,0 +1,137 @@
+"""Тест моделей данных для операторов и смен"""
+import sys
+sys.path.insert(0, 'src')
+
+from datetime import date, datetime
+from domains.models.operator import Operator, ShiftType
+from domains.models.shift import Shift, ShiftTypeEnum
+from domains.models.preference import OperatorPreferences, Preference
+
+
+def test_operator():
+    """Тест модели оператора"""
+    print("=" * 60)
+    print("TEST: Model Operator")
+    print("=" * 60)
+    
+    # Создание оператора
+    operator = Operator(
+        id="OP001",
+        name="Иванов Иван Иванович",
+        department="Склад",
+        email="ivanov@company.com",
+        skills=["Morning", "Afternoon", "Evening"],
+        preferred_days=[1, 2, 3, 4, 5],
+        max_shifts_per_week=4,
+        max_weekly_hours=32
+    )
+    
+    print(f"\nOperator: {operator}")
+    print(f"ID: {operator.id}")
+    print(f"Name: {operator.name}")
+    print(f"Department: {operator.department}")
+    print(f"Skills: {operator.skills}")
+    print(f"Preferred days: {operator.preferred_days}")
+    print(f"Can work 'Night' shift: {operator.can_work_shift(ShiftType.NIGHT)}")
+    
+    # Проверка доступности
+    availability = operator.get_availability()
+    print(f"\nAvailability info:")
+    print(f"  Total shifts assigned: {availability['total_shifts_assigned']}")
+    print(f"  Available for assignment: {availability['available_for_assignment']}")
+
+
+def test_shift():
+    """Тест модели смены"""
+    print("\n" + "=" * 60)
+    print("TEST: Model Shift")
+    print("=" * 60)
+    
+    # Создание смены
+    shift_date = date(2024, 12, 9)  # Пятница
+    shift = Shift(
+        date=shift_date,  # Используем datetime.date объект
+        shift_type=ShiftTypeEnum.AFTERNOON,
+        start_time="14:00",
+        end_time="22:00",
+        department_required="Склад",
+        required_skills=["Evening"],
+        min_operators=2,
+        max_operators=4,
+        description="Вечерняя смена на складе"
+    )
+    
+    print(f"\nShift: {shift}")
+    print(f"Date: {shift.date}")
+    print(f"Type: {shift.shift_type}")
+    print(f"Start time: {shift.start_time}")
+    print(f"End time: {shift.end_time}")
+    print(f"Duration (hours): {shift.duration_hours}")
+    # Проверка совместимости
+    op = Operator(
+        id="OP001",
+        name="Иванов",
+        skills=["Afternoon"]
+    )
+    
+    is_available = shift.is_available(op)
+    print(f"\nOperator {op.name} available for this shift: {is_available}")
+
+
+def test_preferences():
+    """Тест модели предпочтений"""
+    print("\n" + "=" * 60)
+    print("TEST: Model Preferences")
+    print("=" * 60)
+    
+    # Создание предпочтений
+    prefs = OperatorPreferences(
+        operator_id="OP001",
+        shift_preferences=[],
+        day_preferences=[],
+        time_preferences=[],
+        rest_preferences=[],
+        sequence_preferences=[]
+    )
+    
+    # Добавление предпочтений
+    prefs.add_shift_preference(
+        ShiftType.MORNING,
+        strength=80,
+        reason="Хочу работать утром"
+    )
+    
+    prefs.add_day_preference(
+        {1, 2, 3},  # Пн-Ср
+        strength=70,
+        reason="Лучше работать в начале недели"
+    )
+    
+    prefs.add_time_preference(
+        ShiftType.MORNING,
+        strength=75,
+        reason="Вечерная работа - не хочу"
+    )
+    
+    print(f"\nNumber of shift preferences: {len(prefs.shift_preferences)}")
+    print(f"Number of day preferences: {len(prefs.day_preferences)}")
+    # Получение лучших предпочтений
+    best_shift = prefs.get_best_shift_type()
+    best_days = prefs.get_best_days()
+    
+    print(f"Best shift type: {best_shift}")
+    print(f"Best days: {best_days}")
+    
+def test_all():
+    """Запуск всех тестов"""
+    test_operator()
+    test_shift()
+    test_preferences()
+    
+    print("\n" + "=" * 60)
+    print("All tests passed! ✓")
+    print("=" * 60)
+
+
+if __name__ == "__main__":
+    test_all()

tests/unit/__init__.py

@@ -0,0 +1 @@
+"""Unit tests package."""