Update test_scrapers.py

sakan811 · Jun 16, 2024 · 468d4a1 · 468d4a1
1 parent 6d0cabd
commit 468d4a1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_scrapers.py b/tests/test_scrapers.py
@@ -59,7 +59,7 @@ def test_check_if_all_date_was_scraped_csv() -> None:
     )
 
     thread_scrape = ThreadPoolScraper(hotel_stay)
-    thread_scrape.thread_scrape(timezone=city_timezone, max_workers=1)
+    thread_scrape.thread_scrape(timezone=city_timezone, max_workers=3)
     check_csv_if_all_date_was_scraped()
 
     with sqlite3.connect(sqlite_name) as conn:
@@ -103,7 +103,7 @@ def test_thread_scraper() -> None:
     )
 
     thread_scrape = ThreadPoolScraper(hotel_stay)
-    df = thread_scrape.thread_scrape(timezone=city_timezone)
+    df = thread_scrape.thread_scrape(timezone=city_timezone, max_workers=3)
 
     assert not df.empty
 
@@ -134,7 +134,7 @@ def test_thread_scraper_past_month() -> None:
     )
 
     thread_scrape = ThreadPoolScraper(hotel_stay)
-    df = thread_scrape.thread_scrape(timezone=city_timezone)
+    df = thread_scrape.thread_scrape(timezone=city_timezone, max_workers=3)
 
     assert df is None
 
@@ -268,7 +268,7 @@ def test_check_if_all_date_was_scraped() -> None:
     )
 
     thread_scrape = ThreadPoolScraper(hotel_stay)
-    thread_scrape.thread_scrape(to_sqlite=True, timezone=city_timezone)
+    thread_scrape.thread_scrape(to_sqlite=True, timezone=city_timezone, max_workers=3)
     check_db_if_all_date_was_scraped(sqlite_name)
 
     with sqlite3.connect(sqlite_name) as conn: