Skip to content

Commit cb6d937

Browse files
committed
add fixtures for migration tests
1 parent 419edcb commit cb6d937

1 file changed

Lines changed: 27 additions & 0 deletions

File tree

tests/conftest.py

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,8 @@
3434
from pathlib import Path
3535
import pytest
3636
import openml_sklearn
37+
from openml._api import HTTPClient, MinIOClient
38+
from openml.enums import APIVersion
3739

3840
import openml
3941
from openml.testing import TestBase
@@ -307,3 +309,28 @@ def workdir(tmp_path):
307309
os.chdir(tmp_path)
308310
yield tmp_path
309311
os.chdir(original_cwd)
312+
313+
314+
@pytest.fixture
315+
def use_api_v1() -> None:
316+
openml.config.set_api_version(api_version=APIVersion.V1)
317+
318+
319+
@pytest.fixture
320+
def use_api_v2() -> None:
321+
openml.config.set_api_version(api_version=APIVersion.V2)
322+
323+
324+
@pytest.fixture
325+
def http_client_v1() -> HTTPClient:
326+
return HTTPClient(api_version=APIVersion.V1)
327+
328+
329+
@pytest.fixture
330+
def http_client_v2() -> HTTPClient:
331+
return HTTPClient(api_version=APIVersion.V2)
332+
333+
334+
@pytest.fixture
335+
def minio_client() -> MinIOClient:
336+
return MinIOClient()

0 commit comments

Comments
 (0)