Skip to content

Commit b4e052d

Browse files
committed
added unit testcase for flow tagging #214
1 parent be9dd3d commit b4e052d

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tests/test_flows/test_flow.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,6 +198,7 @@ def test_sklearn_to_upload_to_flow(self):
198198
estimator=model, param_distributions=parameter_grid, cv=cv)
199199
rs.fit(X, y)
200200
flow = openml.flows.sklearn_to_flow(rs)
201+
flow.tags.extend(['openml-python', 'unittest'])
201202

202203
# Add the sentinel to all name strings in all subflows. Adds it to
203204
# name to make it easier in the web gui to see that the flow is only
@@ -253,5 +254,6 @@ def test_sklearn_to_upload_to_flow(self):
253254
% sentinel
254255

255256
self.assertEqual(new_flow.name, fixture_name)
256-
257+
self.assertTrue('openml-python' in new_flow.tags)
258+
self.assertTrue('unittest' in new_flow.tags)
257259
new_flow.model.fit(X, y)

0 commit comments

Comments
 (0)