Fix linting errors: remove unused imports, prefix unused variables, fix whitespace

Author: Prachig-Microsoft

src/tests/ContentProcessorWorkflow/services/test_content_process_models.py

@@ -17,7 +17,7 @@
 
 class TestArtifactType:
     """Test ArtifactType enum"""
-    
+
     def test_artifact_type_values(self):
         """Test all artifact type enum values"""
         assert ArtifactType.Undefined == "undefined"
@@ -31,7 +31,7 @@ def test_artifact_type_values(self):
 
 class TestPipelineStep:
     """Test PipelineStep enum"""
-    
+
     def test_pipeline_step_values(self):
         """Test all pipeline step enum values"""
         assert PipelineStep.Transform == "transform"
@@ -43,7 +43,7 @@ def test_pipeline_step_values(self):
 
 class TestProcessFile:
     """Test ProcessFile model"""
-    
+
     def test_process_file_creation(self):
         """Test creating a ProcessFile instance"""
         file = ProcessFile(
@@ -55,15 +55,15 @@ def test_process_file_creation(self):
             artifact_type=ArtifactType.SourceContent,
             processed_by="system"
         )
-        
+
         assert file.process_id == "proc-123"
         assert file.id == "file-456"
         assert file.name == "test.pdf"
         assert file.size == 1024
         assert file.mime_type == "application/pdf"
         assert file.artifact_type == ArtifactType.SourceContent
         assert file.processed_by == "system"
-    
+
     def test_process_file_serialization(self):
         """Test ProcessFile JSON serialization"""
         file = ProcessFile(
@@ -75,15 +75,15 @@ def test_process_file_serialization(self):
             artifact_type=ArtifactType.SourceContent,
             processed_by="system"
         )
-        
+
         data = file.model_dump()
         assert data["process_id"] == "proc-123"
         assert data["artifact_type"] == "source_content"
 
 
 class TestPipelineStatus:
     """Test PipelineStatus model"""
-    
+
     def test_pipeline_status_creation(self):
         """Test creating a PipelineStatus instance"""
         now = datetime.now()
@@ -98,7 +98,7 @@ def test_pipeline_status_creation(self):
             remaining_steps=["evaluate"],
             completed_steps=["extract"]
         )
-        
+
         assert status.process_id == "proc-123"
         assert status.schema_id == "schema-1"
         assert status.metadata_id == "meta-1"
@@ -107,7 +107,7 @@ def test_pipeline_status_creation(self):
         assert status.steps == ["extract", "map"]
         assert status.remaining_steps == ["evaluate"]
         assert status.completed_steps == ["extract"]
-    
+
     def test_pipeline_status_defaults(self):
         """Test PipelineStatus default values"""
         now = datetime.now()
@@ -117,7 +117,7 @@ def test_pipeline_status_defaults(self):
             metadata_id="meta-1",
             creation_time=now
         )
-        
+
         assert status.completed is False
         assert status.last_updated_time is None
         assert status.steps == []
@@ -127,11 +127,11 @@ def test_pipeline_status_defaults(self):
 
 class TestContentProcessMessage:
     """Test ContentProcessMessage model"""
-    
+
     def test_content_process_message_creation(self):
         """Test creating a ContentProcessMessage instance"""
         now = datetime.now()
-        
+
         file = ProcessFile(
             process_id="proc-123",
             id="file-456",
@@ -141,54 +141,54 @@ def test_content_process_message_creation(self):
             artifact_type=ArtifactType.SourceContent,
             processed_by="system"
         )
-        
+
         status = PipelineStatus(
             process_id="proc-123",
             schema_id="schema-1",
             metadata_id="meta-1",
             creation_time=now
         )
-        
+
         message = ContentProcessMessage(
             process_id="proc-123",
             files=[file],
             pipeline_status=status
         )
-        
+
         assert message.process_id == "proc-123"
         assert len(message.files) == 1
         assert message.files[0].name == "test.pdf"
         assert message.pipeline_status.schema_id == "schema-1"
-    
+
     def test_content_process_message_defaults(self):
         """Test ContentProcessMessage default values"""
         now = datetime.now()
-        
+
         # pipeline_status requires certain fields, so we provide them
         status = PipelineStatus(
             process_id="proc-123",
             schema_id="schema-1",
             metadata_id="meta-1",
             creation_time=now
         )
-        
+
         message = ContentProcessMessage(
             process_id="proc-123",
             pipeline_status=status
         )
-        
+
         assert message.process_id == "proc-123"
         assert message.files == []
         assert message.pipeline_status.process_id == "proc-123"
 
 
 class TestContentProcessRecord:
     """Test ContentProcessRecord model"""
-    
+
     def test_content_process_record_creation(self):
         """Test creating a ContentProcessRecord instance"""
         now = datetime.now()
-        
+
         record = ContentProcessRecord(
             id="rec-123",
             process_id="proc-123",
@@ -202,7 +202,7 @@ def test_content_process_record_creation(self):
             result={"key": "value"},
             confidence={"score": 0.9}
         )
-        
+
         assert record.id == "rec-123"
         assert record.process_id == "proc-123"
         assert record.processed_file_name == "test.pdf"
@@ -211,46 +211,46 @@ def test_content_process_record_creation(self):
         assert record.entity_score == 0.95
         assert record.schema_score == 0.90
         assert record.result == {"key": "value"}
-    
+
     def test_content_process_record_defaults(self):
         """Test ContentProcessRecord default values"""
         record = ContentProcessRecord(id="rec-123")
-        
+
         assert record.process_id == ""
         assert record.processed_file_name is None
         assert record.processed_file_mime_type is None
         assert record.entity_score == 0.0
         assert record.schema_score == 0.0
-    
+
     def test_to_cosmos_dict(self):
         """Test ContentProcessRecord.to_cosmos_dict method"""
         now = datetime.now()
-        
+
         record = ContentProcessRecord(
             id="rec-123",
             process_id="proc-123",
             processed_file_name="test.pdf",
             imported_time=now,
             status="completed"
         )
-        
+
         cosmos_dict = record.to_cosmos_dict()
-        
+
         assert cosmos_dict["id"] == "rec-123"
         assert cosmos_dict["process_id"] == "proc-123"
         assert cosmos_dict["processed_file_name"] == "test.pdf"
         assert cosmos_dict["status"] == "completed"
         # imported_time should remain as datetime object, not converted to string
         assert isinstance(cosmos_dict.get("imported_time"), datetime)
-    
+
     def test_extra_fields_allowed(self):
         """Test that ContentProcessRecord allows extra fields"""
         record = ContentProcessRecord(
             id="rec-123",
             process_id="proc-123",
             extra_field="extra_value"
         )
-        
+
         # Extra fields should be preserved in model_dump
         data = record.model_dump()
         assert data.get("extra_field") == "extra_value"