11# License: BSD 3-Clause
22from __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
86from openml .testing import TestAPIBase
97from 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):
6051class 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