Skip to content

Commit 9e5db43

Browse files
update client confit in test_evaluation_measure
1 parent a19cb35 commit 9e5db43

1 file changed

Lines changed: 23 additions & 38 deletions

File tree

tests/test_api/test_evaluation_measures.py renamed to tests/test_api/test_evaluation_measure.py

Lines changed: 23 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -1,30 +1,20 @@
11
# License: BSD 3-Clause
22
from __future__ import annotations
33

4-
import pytest
5-
from openml._api.config import settings
6-
7-
from openml._api.resources.evaluation_measures import EvaluationMeasuresV1, EvaluationMeasuresV2
4+
import pytest
5+
from openml._api.resources.evaluation_measure import EvaluationMeasureV1API, EvaluationMeasureV2API
86
from openml.testing import TestAPIBase
97
from openml._api.resources.base.fallback import FallbackProxy
108

119

12-
class TestEvaluationMeasuresV1(TestAPIBase):
10+
class TestEvaluationMeasureV1(TestAPIBase):
1311
"""Tests for V1 XML API implementation of evaluation measures."""
1412

1513
_multiprocess_can_split_ = True
1614

1715
def setUp(self) -> None:
1816
super().setUp()
19-
self.client = self._get_http_client(
20-
server=settings.api.v1.server,
21-
base_url=settings.api.v1.base_url,
22-
api_key=settings.api.v1.api_key,
23-
timeout=settings.api.v1.timeout,
24-
retries=settings.connection.retries,
25-
retry_policy=settings.connection.retry_policy,
26-
)
27-
self.resource = EvaluationMeasuresV1(self.client)
17+
self.resource = EvaluationMeasureV1API(self.http_client)
2818

2919
@pytest.mark.uses_test_server()
3020
def test_list(self):
@@ -33,22 +23,23 @@ def test_list(self):
3323
assert all(isinstance(s, str) for s in measures) is True
3424

3525

36-
class TestEvaluationMeasuresV2(TestAPIBase):
26+
class TestEvaluationMeasureV2(TestAPIBase):
3727
"""Tests for V2 JSON API implementation of evaluation measures."""
3828

3929
_multiprocess_can_split_ = True
4030

4131
def setUp(self) -> None:
4232
super().setUp()
4333
self.client = self._get_http_client(
44-
server=settings.api.v2.server,
45-
base_url=settings.api.v2.base_url,
46-
api_key=settings.api.v2.api_key,
47-
timeout=settings.api.v2.timeout,
48-
retries=settings.connection.retries,
49-
retry_policy=settings.connection.retry_policy,
34+
server="http://localhost:8001/",
35+
base_url="",
36+
api_key="",
37+
timeout_seconds=self.timeout_seconds,
38+
retries=self.retries,
39+
retry_policy=self.retry_policy,
40+
cache=self.cache,
5041
)
51-
self.resource = EvaluationMeasuresV2(self.client)
42+
self.resource = EvaluationMeasureV2API(self.client)
5243

5344
@pytest.mark.uses_test_server()
5445
def test_list(self):
@@ -60,24 +51,18 @@ def test_list(self):
6051
class TestEvaluationMeasuresCombined(TestAPIBase):
6152
def setUp(self):
6253
super().setUp()
63-
self.v1_client = self._get_http_client(
64-
server=settings.api.v1.server,
65-
base_url=settings.api.v1.base_url,
66-
api_key=settings.api.v1.api_key,
67-
timeout=settings.api.v1.timeout,
68-
retries=settings.connection.retries,
69-
retry_policy=settings.connection.retry_policy,
70-
)
54+
self.v1_client = self.http_client
7155
self.v2_client = self._get_http_client(
72-
server=settings.api.v2.server,
73-
base_url=settings.api.v2.base_url,
74-
api_key=settings.api.v2.api_key,
75-
timeout=settings.api.v2.timeout,
76-
retries=settings.connection.retries,
77-
retry_policy=settings.connection.retry_policy,
56+
server="http://localhost:8001/",
57+
base_url="",
58+
api_key="",
59+
timeout_seconds=self.timeout_seconds,
60+
retries=self.retries,
61+
retry_policy=self.retry_policy,
62+
cache=self.cache,
7863
)
79-
self.resource_v1 = EvaluationMeasuresV1(self.v1_client)
80-
self.resource_v2 = EvaluationMeasuresV2(self.v2_client)
64+
self.resource_v1 = EvaluationMeasureV1API(self.v1_client)
65+
self.resource_v2 = EvaluationMeasureV2API(self.v2_client)
8166
self.resource_fallback = FallbackProxy(self.resource_v2, self.resource_v1)
8267

8368
@pytest.mark.uses_test_server()

0 commit comments

Comments
 (0)