Skip to content

Commit 21e0007

Browse files
committed
fix test for dataset tagging
1 parent 90fab53 commit 21e0007

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

tests/test_datasets/test_dataset.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import numpy as np
22
from scipy import sparse
33
import six
4+
from time import time
45

56
from openml.testing import TestBase
67
import openml
@@ -97,13 +98,13 @@ def setUp(self):
9798
self.dataset = openml.datasets.get_dataset(125)
9899

99100
def test_tagging(self):
100-
tag = "testing_tag{}".format(self.id)
101+
tag = "testing_tag_{}_{}".format(self.id(), time())
101102
ds_list = openml.datasets.list_datasets(tag=tag)
102103
self.assertEqual(len(ds_list), 0)
103104
self.dataset.push_tag(tag)
104105
ds_list = openml.datasets.list_datasets(tag=tag)
105106
self.assertEqual(len(ds_list), 1)
106-
self.assertEqual(ds_list[0]['did'], 125)
107+
self.assertIn(125, ds_list)
107108
self.dataset.remove_tag(tag)
108109
ds_list = openml.datasets.list_datasets(tag=tag)
109110
self.assertEqual(len(ds_list), 0)

0 commit comments

Comments
 (0)