fix tests

Author: soodoku

tests/test_archive_batch.py

@@ -9,7 +9,6 @@
 import os
 import tempfile
 import unittest
-from datetime import datetime
 
 from piedomains import DomainClassifier
 from piedomains.fetchers import ArchiveFetcher

tests/test_data_collector.py

@@ -10,8 +10,7 @@
 from pathlib import Path
 from unittest.mock import MagicMock, patch
 
-from piedomains import DataCollector, DomainClassifier
-from piedomains import TextClassifier, ImageClassifier
+from piedomains import DataCollector, DomainClassifier, TextClassifier
 
 
 class TestDataCollector(unittest.TestCase):

tests/test_integration.py

@@ -17,9 +17,9 @@
 
 from piedomains import DomainClassifier, classify_domains
 from piedomains.archive_org_downloader import download_from_archive_org, get_urls_year
+from piedomains.content_processor import ContentProcessor
 from piedomains.fetchers import ArchiveFetcher, PlaywrightFetcher, get_fetcher
 from piedomains.piedomains_logging import configure_logging, get_logger
-from piedomains.processors.content_processor import ContentProcessor
 from piedomains.utils import is_within_directory
 
 # Configure test logging

tests/test_performance.py

@@ -16,7 +16,7 @@
 import pytest
 
 from piedomains.api import DomainClassifier
-from piedomains.processors.text_processor import TextProcessor
+from piedomains.text_processor import TextProcessor
 
 
 class TestPerformanceBenchmarks(unittest.TestCase):
@@ -256,7 +256,7 @@ def test_cache_directory_creation(self):
         DomainClassifier(cache_dir=new_cache_dir)
 
         # The directory should be created when processors are initialized
-        from piedomains.processors.content_processor import ContentProcessor
+        from piedomains.content_processor import ContentProcessor
 
         processor = ContentProcessor(new_cache_dir)