fix tests

apify · Aug 22, 2024 · 4dd0a6b · 4dd0a6b
1 parent ea47b33
commit 4dd0a6b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/unit/_autoscaling/test_snapshotter.py b/tests/unit/_autoscaling/test_snapshotter.py
@@ -174,7 +174,7 @@ def test_memory_load_evaluation_logs_warning_on_high_usage(
     snapshotter: Snapshotter,
 ) -> None:
     mock_logger_warn = MagicMock()
-    monkeypatch.setattr(getLogger('crawlee.autoscaling.snapshotter'), 'warning', mock_logger_warn)
+    monkeypatch.setattr(getLogger('crawlee._autoscaling.snapshotter'), 'warning', mock_logger_warn)
     snapshotter._max_memory_size = ByteSize.from_gb(8)
 
     high_memory_usage = ByteSize.from_gb(8) * 0.9  # 90% of 8 GB

diff --git a/tests/unit/parsel_crawler/test_parsel_crawler.py b/tests/unit/parsel_crawler/test_parsel_crawler.py
@@ -215,7 +215,7 @@ def test_import_error_handled() -> None:
     with mock.patch.dict('sys.modules', {'parsel': None}):
         # Invalidate ParselCrawler import
         sys.modules.pop('crawlee.parsel_crawler', None)
-        sys.modules.pop('crawlee.parsel_crawler.parsel_crawler', None)
+        sys.modules.pop('crawlee.parsel_crawler._parsel_crawler', None)
 
         with pytest.raises(ImportError) as import_error:
             from crawlee.parsel_crawler import ParselCrawler  # noqa: F401