Skip to content

Commit 6dce274

Browse files
committed
added a unit test to check on the size of the list results (prevents server bugs)
1 parent a7d9ed4 commit 6dce274

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

tests/test_setups/test_setup_functions.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@ def get_params(self, deep=True):
4747
return {}
4848

4949

50-
5150
class TestRun(TestBase):
5251

5352
def test_nonexisting_setup_exists(self):
@@ -118,7 +117,6 @@ def test_get_setup(self):
118117
else:
119118
self.assertEquals(len(current.parameters), num_params[idx])
120119

121-
122120
def test_setup_list_filter_flow(self):
123121
openml.config.server = self.production_server
124122

@@ -129,3 +127,17 @@ def test_setup_list_filter_flow(self):
129127
self.assertGreater(len(setups), 0) # TODO: please adjust 0
130128
for setup_id in setups.keys():
131129
self.assertEquals(setups[setup_id].flow_id, flow_id)
130+
131+
def test_setuplist_offset(self):
132+
# TODO: remove after pull on live for better testing
133+
# openml.config.server = self.production_server
134+
135+
size = 100
136+
setups = openml.setups.list_setups(offset=0, size=size)
137+
self.assertEquals(len(setups), size)
138+
setups2 = openml.setups.list_setups(offset=size, size=size)
139+
self.assertEquals(len(setups), size)
140+
141+
all = set(setups.keys()).union(setups2.keys())
142+
143+
self.assertEqual(len(all), size * 2)

0 commit comments

Comments
 (0)