fix: Remove unused imports and variables in checkpoint tests
- 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:
parent
280fb3fefe
commit
b293dc691d
1 changed files with 4 additions and 5 deletions
|
|
@ -26,8 +26,7 @@ Tests cover:
|
|||
"""
|
||||
|
||||
import pytest
|
||||
from unittest.mock import Mock, MagicMock
|
||||
from datetime import datetime
|
||||
from unittest.mock import Mock
|
||||
|
||||
|
||||
class TestCheckpointCreation:
|
||||
|
|
@ -320,7 +319,7 @@ class TestIntegrationScenarios:
|
|||
mock_checkpoint.total_documents = 3
|
||||
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_type="raptor",
|
||||
doc_ids=["doc1", "doc2", "doc3"],
|
||||
|
|
@ -345,7 +344,7 @@ class TestIntegrationScenarios:
|
|||
mock_checkpoint.id = "checkpoint_123"
|
||||
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_type="raptor",
|
||||
doc_ids=["doc1", "doc2", "doc3"],
|
||||
|
|
@ -386,7 +385,7 @@ class TestIntegrationScenarios:
|
|||
mock_checkpoint.id = "checkpoint_123"
|
||||
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_type="raptor",
|
||||
doc_ids=["doc1", "doc2", "doc3", "doc4", "doc5"],
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue