@@ -136,7 +136,7 @@ def test_get_downloaded_packages_for_real_python_packages(temp_directory):
136136 Assert that for genuine pypi packages, get_downloaded_packages() actually picks up the correct package names.
137137 To avoid flakiness, we only check for package names.
138138 """
139- packages = ["requests" , "numpy" , "pandas" ]
139+ packages = ["requests" , "numpy" ]
140140 target_folder = os .path .join (temp_directory , "packages" )
141141 pip_install_packages_to_target_folder (packages , target_folder )
142142 for package in packages :
@@ -148,9 +148,9 @@ def test_get_downloaded_packages_for_real_python_packages(temp_directory):
148148 )
149149 package_names = {package .name for package in downloaded_packages_dict .keys ()}
150150 for package_name in packages + [
151- "six " ,
152- "pytz " ,
153- ]: # six and pytz are integral dependencies
151+ "certifi " ,
152+ "urllib3 " ,
153+ ]: # certifi and urllib3 are integral dependencies
154154 assert package_name in package_names
155155
156156
@@ -162,7 +162,7 @@ def test_pip_timeout(temp_directory):
162162 """
163163 Assert that timeout parameter works fine.
164164 """
165- packages = ["requests" , "numpy" , "pandas" ]
165+ packages = ["requests" , "numpy" ]
166166 target_folder = os .path .join (temp_directory , "packages" )
167167 with pytest .raises (TimeoutExpired ):
168168 pip_install_packages_to_target_folder (packages , target_folder , timeout = 1 )
@@ -295,7 +295,7 @@ def test_identify_supported_packages_all_cases():
295295 reason = "Subprocess calls are not allowed within stored procedures" ,
296296)
297297def test_valid_pip_install (temp_directory ):
298- packages = ["requests" , "numpy" , "pandas" ]
298+ packages = ["requests" , "numpy" ]
299299 target_folder = os .path .join (temp_directory , "packages" )
300300 pip_install_packages_to_target_folder (packages , target_folder )
301301 for package in packages :
0 commit comments