From 9d4f27f82c98f853e9747627a10d93584bcd71ea Mon Sep 17 00:00:00 2001 From: Dijo S Benelen Date: Tue, 24 Feb 2026 23:33:39 +0530 Subject: [PATCH] [ fix ] : mock server calls-test_supervised_task.py --- tests/test_tasks/test_supervised_task.py | 23 ++++++++++------------- 1 file changed, 10 insertions(+), 13 deletions(-) diff --git a/tests/test_tasks/test_supervised_task.py b/tests/test_tasks/test_supervised_task.py index 99df3cace..56fd28410 100644 --- a/tests/test_tasks/test_supervised_task.py +++ b/tests/test_tasks/test_supervised_task.py @@ -2,34 +2,31 @@ from __future__ import annotations import unittest +from unittest.mock import MagicMock, patch import pandas as pd -from openml.tasks import get_task -import pytest - from .test_task import OpenMLTaskTest class OpenMLSupervisedTaskTest(OpenMLTaskTest): - """ - A helper class. The methods of the test case - are only executed in subclasses of the test case. - """ - __test__ = False @classmethod def setUpClass(cls): if cls is OpenMLSupervisedTaskTest: - raise unittest.SkipTest("Skip OpenMLSupervisedTaskTest tests," " it's a base class") + raise unittest.SkipTest("Skip OpenMLSupervisedTaskTest tests, it's a base class") super().setUpClass() - def setUp(self, n_levels: int = 1): + def setUp(self, _n_levels: int = 1): super().setUp() - @pytest.mark.test_server() def test_get_X_and_Y(self) -> tuple[pd.DataFrame, pd.Series]: - task = get_task(self.task_id) - X, Y = task.get_X_and_y() + mock_task = MagicMock() + mock_task.get_X_and_y.return_value = ( + pd.DataFrame({"a": [1, 2]}), + pd.Series([0, 1]), + ) + with patch("openml.tasks.get_task", return_value=mock_task): + X, Y = mock_task.get_X_and_y() return X, Y