gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: Remove unused imports and variables in checkpoint tests

Browse source 
- Remove unused MagicMock import
- Remove unused datetime import
- Remove unused checkpoint variables in integration tests
- All 22 tests still passing
- Ruff linting now passes
This commit is contained in:
hsparks.codes 2025-12-03 09:33:51 +01:00
parent 280fb3fefe
commit b293dc691d
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
test/unit_test/services/test_checkpoint_service.py
View file

@ -26,8 +26,7 @@ Tests cover:
""" """
import pytest import pytest
from unittest.mock import Mock, MagicMock from unittest.mock import Mock
from datetime import datetime
class TestCheckpointCreation: class TestCheckpointCreation:
@ -320,7 +319,7 @@ class TestIntegrationScenarios:
mock_checkpoint.total_documents = 3 mock_checkpoint.total_documents = 3
mock_checkpoint_service.create_checkpoint.return_value = mock_checkpoint mock_checkpoint_service.create_checkpoint.return_value = mock_checkpoint
checkpoint = mock_checkpoint_service.create_checkpoint( mock_checkpoint_service.create_checkpoint(
task_id="task_123", task_id="task_123",
task_type="raptor", task_type="raptor",
doc_ids=["doc1", "doc2", "doc3"], doc_ids=["doc1", "doc2", "doc3"],
@ -345,7 +344,7 @@ class TestIntegrationScenarios:
mock_checkpoint.id = "checkpoint_123" mock_checkpoint.id = "checkpoint_123"
mock_checkpoint_service.create_checkpoint.return_value = mock_checkpoint mock_checkpoint_service.create_checkpoint.return_value = mock_checkpoint
checkpoint = mock_checkpoint_service.create_checkpoint( mock_checkpoint_service.create_checkpoint(
task_id="task_123", task_id="task_123",
task_type="raptor", task_type="raptor",
doc_ids=["doc1", "doc2", "doc3"], doc_ids=["doc1", "doc2", "doc3"],
@ -386,7 +385,7 @@ class TestIntegrationScenarios:
mock_checkpoint.id = "checkpoint_123" mock_checkpoint.id = "checkpoint_123"
mock_checkpoint_service.create_checkpoint.return_value = mock_checkpoint mock_checkpoint_service.create_checkpoint.return_value = mock_checkpoint
checkpoint = mock_checkpoint_service.create_checkpoint( mock_checkpoint_service.create_checkpoint(
task_id="task_123", task_id="task_123",
task_type="raptor", task_type="raptor",
doc_ids=["doc1", "doc2", "doc3", "doc4", "doc5"], doc_ids=["doc1", "doc2", "doc3", "doc4", "doc5"],