Skip to content

Commit 33a34e1

Browse files
committed
extended unit test
1 parent 06d1ad8 commit 33a34e1

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/test_datasets/test_dataset_functions.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818

1919
from openml.datasets.functions import (_get_cached_dataset,
2020
_get_cached_dataset_features,
21+
_get_cached_dataset_qualities,
2122
_get_cached_datasets,
2223
_get_dataset_description,
2324
_get_dataset_arff,
@@ -63,11 +64,13 @@ def test__get_cached_dataset(self, ):
6364
openml.config.set_cache_directory(self.static_cache_dir)
6465
dataset = _get_cached_dataset(2)
6566
features = _get_cached_dataset_features(2)
67+
qualities = _get_cached_dataset_qualities(2)
6668
self.assertIsInstance(dataset, OpenMLDataset)
6769
self.assertTrue(len(dataset.features) > 0)
6870
self.assertTrue(len(dataset.features) == len(features['oml:feature']))
71+
self.assertTrue(len(dataset.qualities) == len(qualities['oml:quality']))
6972

70-
def test_get_chached_dataset_description(self):
73+
def test_get_cached_dataset_description(self):
7174
openml.config.set_cache_directory(self.static_cache_dir)
7275
description = openml.datasets.functions._get_cached_dataset_description(2)
7376
self.assertIsInstance(description, dict)

0 commit comments

Comments
 (0)