don't hardcode flow number

Author: amueller

tests/test_flows/test_flow.py

@@ -36,8 +36,8 @@ class TestFlow(TestBase):
     _multiprocess_can_split_ = True
 
     def test_get_flow(self):
-        # We need to use the production server here because 4024 is not the test
-        # server
+        # We need to use the production server here because 4024 is not the
+        # test server
         openml.config.server = self.production_server
 
         flow = openml.flows.get_flow(4024)
@@ -68,14 +68,16 @@ def test_get_flow(self):
         self.assertEqual(len(subflow_3.components), 0)
 
     def test_tagging(self):
-        flow = openml.flows.get_flow(4024)
+        flow_list = openml.flows.list_flows(size=1)
+        flow_id = list(flow_list.keys())[0]
+        flow = openml.flows.get_flow(flow_id)
         tag = "testing_tag_{}_{}".format(self.id(), time())
         flow_list = openml.flows.list_flows(tag=tag)
         self.assertEqual(len(flow_list), 0)
         flow.push_tag(tag)
         flow_list = openml.flows.list_flows(tag=tag)
         self.assertEqual(len(flow_list), 1)
-        self.assertIn(4024, flow_list)
+        self.assertIn(flow_id, flow_list)
         flow.remove_tag(tag)
         flow_list = openml.flows.list_flows(tag=tag)
         self.assertEqual(len(flow_list), 0)