fix: great expectations unit tests referring to deprecated method (#1558

) * fix: ge unit tests refering to deprecated method * change cache_file to use requests instead of urllib * fix(linting): code formatting * requests response raise for status * fix(linting): code formatting * change read to content * use cache_file to download zipfile * fix(linting): code formatting --------- Co-authored-by: Luís Portela Afonso <[email protected]> Co-authored-by: Azory YData Bot <[email protected]>
ydataai · Mar 18, 2024 · 8d428f2 · 8d428f2
1 parent 53dd7cb
commit 8d428f2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 13 deletions.
diff --git a/src/ydata_profiling/utils/cache.py b/src/ydata_profiling/utils/cache.py
@@ -1,7 +1,8 @@
 """Dataset cache utility functions"""
 import zipfile
 from pathlib import Path
-from urllib import request
+
+from requests import get as get_file
 
 from ydata_profiling.utils.paths import get_data_path
 
@@ -24,8 +25,10 @@ def cache_file(file_name: str, url: str) -> Path:
 
     # If not exists, download and create file
     if not file_path.exists():
-        response = request.urlopen(url)
-        file_path.write_bytes(response.read())
+        response = get_file(url, allow_redirects=True)
+        response.raise_for_status()
+
+        file_path.write_bytes(response.content)
 
     return file_path
 
@@ -42,16 +45,11 @@ def cache_zipped_file(file_name: str, url: str) -> Path:
     """
 
     data_path = get_data_path()
-    data_path.mkdir(exist_ok=True)
-
     file_path = data_path / file_name
 
     # If not exists, download and create file
     if not file_path.exists():
-        response = request.urlopen(url)
-
-        tmp_path = data_path / "tmp.zip"
-        tmp_path.write_bytes(response.read())
+        tmp_path = cache_file("tmp.zip", url)
 
         with zipfile.ZipFile(tmp_path, "r") as zip_file:
             zip_file.extract(file_path.name, data_path)

diff --git a/tests/unit/test_ge_integration.py b/tests/unit/test_ge_integration.py
@@ -42,7 +42,7 @@ def test_to_expectations_suite_context_save_and_build_data_docs(mod, context, df
     mod.data_context.DataContext.assert_not_called()
     mod.dataset.PandasDataset.assert_called_once()
 
-    context.create_expectation_suite.assert_called_once()
+    context.add_expectation_suite.assert_called_once()
     context.save_expectation_suite.assert_called_once()
     context.build_data_docs.assert_called_once()
     context.open_data_docs.assert_called_once()
@@ -60,7 +60,7 @@ def test_to_expectations_suite_context_no_save_and_build_data_docs(mod, context,
     mod.data_context.DataContext.assert_not_called()
     mod.dataset.PandasDataset.assert_called_once()
 
-    context.create_expectation_suite.assert_called_once()
+    context.add_expectation_suite.assert_called_once()
     context.save_expectation_suite.assert_called_once()
     context.build_data_docs.assert_called_once()
     context.open_data_docs.assert_called_once()
@@ -78,7 +78,7 @@ def test_to_expectations_suite_context_no_save_and_no_build_data_docs(mod, conte
     mod.data_context.DataContext.assert_not_called()
     mod.dataset.PandasDataset.assert_called_once()
 
-    context.create_expectation_suite.assert_called_once()
+    context.add_expectation_suite.assert_called_once()
     context.save_expectation_suite.assert_not_called()
     context.build_data_docs.assert_not_called()
     context.open_data_docs.assert_not_called()
@@ -92,7 +92,7 @@ def test_to_expectations_suite_title(context, df):
         run_validation=False,
     )
 
-    context.create_expectation_suite.assert_called_once_with(
+    context.add_expectation_suite.assert_called_once_with(
         "expectations-dataset", overwrite_existing=True
     )