added final unit tests

Author: janvanrijn

openml/evaluations/functions.py

@@ -71,7 +71,7 @@ def _list_evaluations(api_call):
     if isinstance(evals_dict['oml:evaluations']['oml:evaluation'], list):
         evals_list = evals_dict['oml:evaluations']['oml:evaluation']
     elif isinstance(evals_dict['oml:evaluations']['oml:evaluation'], dict):
-        evals_list = [evals_dict['oml:runs']['oml:run']]
+        evals_list = [evals_dict['oml:evaluations']['oml:evaluation']]
     else:
         raise TypeError()
 

tests/test_evaluations/test_evaluation_functions.py

@@ -1,4 +1,3 @@
-import unittest
 import openml
 import openml.evaluations
 from openml.testing import TestBase
@@ -29,6 +28,42 @@ def test_evaluation_list_filter_uploader(self):
         #     self.assertEquals(evaluations[run_id].uploader, uploader_id)
 
 
+    def test_evaluation_list_filter_uploader(self):
+        openml.config.server = self.production_server
+
+        setup_id = 10
+
+        evaluations = openml.evaluations.list_evaluations("predictive_accuracy", setup=[setup_id])
+
+        self.assertGreater(len(evaluations), 100)
+        for run_id in evaluations.keys():
+            self.assertEquals(evaluations[run_id].setup_id, setup_id)
+
+
+    def test_evaluation_list_filter_flow(self):
+        openml.config.server = self.production_server
+
+        flow_id = 100
+
+        evaluations = openml.evaluations.list_evaluations("predictive_accuracy", flow=[flow_id])
+
+        self.assertGreater(len(evaluations), 2)
+        for run_id in evaluations.keys():
+            self.assertEquals(evaluations[run_id].flow_id, flow_id)
+
+
+    def test_evaluation_list_filter_run(self):
+        openml.config.server = self.production_server
+
+        run_id = 1
+
+        evaluations = openml.evaluations.list_evaluations("predictive_accuracy", id=[run_id])
+
+        self.assertEquals(len(evaluations), 1)
+        for run_id in evaluations.keys():
+            self.assertEquals(evaluations[run_id].run_id, run_id)
+
+
     def test_evaluation_list_limit(self):
         openml.config.server = self.production_server