Skip to content

Commit 85472cc

Browse files
committed
changes requested
1 parent bd8ee24 commit 85472cc

1 file changed

Lines changed: 5 additions & 4 deletions

File tree

tests/test_setups/test_setup_functions.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,9 @@ def test_nonexisting_setup_exists(self):
3838
flow.name = 'TEST%s%s' % (sentinel, flow.name)
3939
flow.publish()
4040

41-
# although the flow exists, we can be sure there are no
42-
# setups (yet) as it hasn't been ran
41+
# although the flow exists (created as of previous statement),
42+
# we can be sure there are no setups (yet) as it was just created
43+
# and hasn't been ran
4344
setup_id = openml.setups.setup_exists(flow, dectree)
4445
self.assertFalse(setup_id)
4546

@@ -87,6 +88,6 @@ def test_setup_get(self):
8788
current = openml.setups.get_setup(setups[idx])
8889
assert current.flow_id > 0
8990
if num_params[idx] == 0:
90-
self.assertTrue(current.parameters is None)
91+
self.assertIsNone(current.parameters)
9192
else:
92-
self.assertTrue(len(current.parameters) == num_params[idx])
93+
self.assertEquals(len(current.parameters), num_params[idx])

0 commit comments

Comments
 (0)