feat: complete production-quality Model Router AI traffic control platform

backend/tests/test_analyzer.py

@@ -0,0 +1,41 @@
+import pytest
+from app.analyzer.heuristics import analyze_request_heuristics, estimate_tokens
+from app.models.schemas import TaskType, PriorityLevel
+
+
+def test_token_estimation():
+    text = "Hello world from Model Router platform!"
+    tokens = estimate_tokens(text)
+    assert tokens > 0
+
+
+def test_heuristic_classification_coding():
+    prompt = "Write a python function to calculate fibonacci sequence using dynamic programming"
+    analysis = analyze_request_heuristics(prompt)
+    assert analysis.task_type == TaskType.CODING
+    assert analysis.coding_required is True
+    assert analysis.complexity >= 0.40
+
+
+def test_heuristic_classification_debugging_complexity():
+    prompt = "Debug this distributed async deadlock issue in the microservices cluster and explain the root cause"
+    analysis = analyze_request_heuristics(prompt)
+    assert analysis.task_type == TaskType.DEBUGGING
+    assert analysis.reasoning_required is True
+    assert analysis.coding_required is True
+    assert analysis.complexity_label == PriorityLevel.HIGH
+    assert analysis.complexity >= 0.75
+
+
+def test_heuristic_classification_summarization():
+    prompt = "Provide a summary and key bullet points of this meeting transcript"
+    analysis = analyze_request_heuristics(prompt)
+    assert analysis.task_type == TaskType.SUMMARIZATION
+    assert analysis.complexity_label in (PriorityLevel.LOW, PriorityLevel.MEDIUM)
+
+
+def test_heuristic_classification_math():
+    prompt = "Calculate the derivative and integral of matrix eigenvalue differential equation"
+    analysis = analyze_request_heuristics(prompt)
+    assert analysis.task_type == TaskType.MATH
+    assert analysis.reasoning_required is True