diff --git a/tests/test_all.py b/tests/test_all.py index d395850..3285611 100644 --- a/tests/test_all.py +++ b/tests/test_all.py @@ -3,15 +3,16 @@ import pandas as pd def test_all(): + sites = [ + "indeed", + "glassdoor", + ] # ziprecruiter/linkedin needs good ip, and temp fix to pass test on ci result = scrape_jobs( - site_name=[ - "indeed", - "glassdoor", - ], # ziprecruiter/linkedin needs good ip, and temp fix to pass test on ci + site_name=sites, search_term="engineer", results_wanted=5, ) assert ( - isinstance(result, pd.DataFrame) and len(result) == 15 + isinstance(result, pd.DataFrame) and len(result) == len(sites) * 5 ), "Result should be a non-empty DataFrame"