Skip to content

Commit 0606243

Browse files
committed
convert test to pytest
1 parent 0560829 commit 0606243

1 file changed

Lines changed: 11 additions & 8 deletions

File tree

tests/test_datasets/test_dataset_functions.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -384,14 +384,6 @@ def test__download_minio_file_works_with_bucket_subdirectory(self):
384384
file_destination
385385
), "_download_minio_file can download from subdirectories"
386386

387-
def test__get_dataset_parquet_not_cached(self):
388-
description = {
389-
"oml:parquet_url": "http://data.openml.org/dataset20/dataset_20.pq",
390-
"oml:id": "20",
391-
}
392-
path = _get_dataset_parquet(description, cache_directory=self.workdir)
393-
assert isinstance(path, Path), "_get_dataset_parquet returns a path"
394-
assert path.is_file(), "_get_dataset_parquet returns path to real file"
395387

396388
@mock.patch("openml._api_calls._download_minio_file")
397389
def test__get_dataset_parquet_is_cached(self, patch):
@@ -1955,6 +1947,17 @@ def test_get_dataset_with_invalid_id() -> None:
19551947
openml.datasets.get_dataset(INVALID_ID)
19561948
assert e.value.code == 111
19571949

1950+
1951+
def test__get_dataset_parquet_not_cached():
1952+
description = {
1953+
"oml:parquet_url": "http://data.openml.org/dataset20/dataset_20.pq",
1954+
"oml:id": "20",
1955+
}
1956+
path = _get_dataset_parquet(description, cache_directory=Path(openml.config.get_cache_directory()))
1957+
assert isinstance(path, Path), "_get_dataset_parquet returns a path"
1958+
assert path.is_file(), "_get_dataset_parquet returns path to real file"
1959+
1960+
19581961
def test_read_features_from_xml_with_whitespace() -> None:
19591962
from openml.datasets.dataset import _read_features
19601963
features_file = Path(__file__).parent.parent / "files" / "misc" / "features_with_whitespaces.xml"

0 commit comments

Comments
 (0)