modified tests

Author: Omswastik-11

tests/test_api/test_run.py

@@ -47,7 +47,7 @@ def test_run_v1_list(run_v1):
     assert "flow_id" in runs_df.columns
 
 
-def test_run_v1_publish_mocked(run_v1, use_api_v1, test_api_key):
+def test_run_v1_publish_mocked(run_v1, test_apikey_v1):
     files = {"description": "<run/>"}
 
     with patch.object(Session, "request") as mock_request:
@@ -66,13 +66,13 @@ def test_run_v1_publish_mocked(run_v1, use_api_v1, test_api_key):
             method="POST",
             url=openml.config.server + "run",
             params={},
-            data={"api_key": test_api_key},
+            data={"api_key": test_apikey_v1},
             headers=openml.config._HEADERS,
             files=files,
         )
 
 
-def test_run_v1_delete_mocked(run_v1, use_api_v1, test_api_key):
+def test_run_v1_delete_mocked(run_v1, test_apikey_v1):
     run_id = 456
 
     with patch.object(Session, "request") as mock_request:
@@ -90,7 +90,7 @@ def test_run_v1_delete_mocked(run_v1, use_api_v1, test_api_key):
         mock_request.assert_called_once_with(
             method="DELETE",
             url=openml.config.server + f"run/{run_id}",
-            params={"api_key": test_api_key},
+            params={"api_key": test_apikey_v1},
             data={},
             headers=openml.config._HEADERS,
             files=None,

tests/test_runs/test_run_functions.py

@@ -1802,7 +1802,7 @@ def test_initialize_model_from_run_nonstrict(self):
         _ = openml.runs.initialize_model_from_run(run_id=1, strict_version=False)
 
 
-@mock.patch.object(requests.Session, "delete")
+@mock.patch.object(requests.Session, "request")
 def test_delete_run_not_owned(mock_delete, test_files_directory, test_server_v1, test_apikey_v1):
     content_file = test_files_directory / "mock_responses" / "runs" / "run_delete_not_owned.xml"
     mock_delete.return_value = create_request_response(
@@ -1817,11 +1817,12 @@ def test_delete_run_not_owned(mock_delete, test_files_directory, test_server_v1,
         openml.runs.delete_run(40_000)
 
     run_url = test_server_v1 + "run/40000"
-    assert run_url == mock_delete.call_args.args[0]
+    assert run_url == mock_delete.call_args.kwargs.get("url")
+    assert "DELETE" == mock_delete.call_args.kwargs.get("method")
     assert test_apikey_v1 == mock_delete.call_args.kwargs.get("params", {}).get("api_key")
 
 
-@mock.patch.object(requests.Session, "delete")
+@mock.patch.object(requests.Session, "request")
 def test_delete_run_success(mock_delete, test_files_directory, test_server_v1, test_apikey_v1):
     content_file = test_files_directory / "mock_responses" / "runs" / "run_delete_successful.xml"
     mock_delete.return_value = create_request_response(
@@ -1833,11 +1834,12 @@ def test_delete_run_success(mock_delete, test_files_directory, test_server_v1, t
     assert success
 
     run_url = test_server_v1 + "run/10591880"
-    assert run_url == mock_delete.call_args.args[0]
+    assert run_url == mock_delete.call_args.kwargs.get("url")
+    assert "DELETE" == mock_delete.call_args.kwargs.get("method")
     assert test_apikey_v1 == mock_delete.call_args.kwargs.get("params", {}).get("api_key")
 
 
-@mock.patch.object(requests.Session, "delete")
+@mock.patch.object(requests.Session, "request")
 def test_delete_unknown_run(mock_delete, test_files_directory, test_server_v1, test_apikey_v1):
     content_file = test_files_directory / "mock_responses" / "runs" / "run_delete_not_exist.xml"
     mock_delete.return_value = create_request_response(
@@ -1852,7 +1854,8 @@ def test_delete_unknown_run(mock_delete, test_files_directory, test_server_v1, t
         openml.runs.delete_run(9_999_999)
 
     run_url = test_server_v1 + "run/9999999"
-    assert run_url == mock_delete.call_args.args[0]
+    assert run_url == mock_delete.call_args.kwargs.get("url")
+    assert "DELETE" == mock_delete.call_args.kwargs.get("method")
     assert test_apikey_v1 == mock_delete.call_args.kwargs.get("params", {}).get("api_key")