@@ -47,7 +47,6 @@ def get_params(self, deep=True):
4747 return {}
4848
4949
50-
5150class 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