feitianyiren
diff --git a/Collapse file
‎openml/study/functions.py‎
Copy file name to clipboard
+43-4Lines changed: 43 additions & 4 deletions b/Collapse file
‎openml/study/functions.py‎
Copy file name to clipboard
+43-4Lines changed: 43 additions & 4 deletions
diff --git a/Collapse file
‎tests/test_evaluations/test_evaluation_functions.py‎
Copy file name to clipboardExpand all lines: tests/test_evaluations/test_evaluation_functions.py
-1Lines changed: 0 additions & 1 deletion b/Collapse file
‎tests/test_evaluations/test_evaluation_functions.py‎
Copy file name to clipboardExpand all lines: tests/test_evaluations/test_evaluation_functions.py
-1Lines changed: 0 additions & 1 deletion
diff --git a/Collapse file
‎tests/test_study/test_study_functions.py‎
Copy file name to clipboard
+16Lines changed: 16 additions & 0 deletions b/Collapse file
‎tests/test_study/test_study_functions.py‎
Copy file name to clipboard
+16Lines changed: 16 additions & 0 deletions
@@ -1,9 +1,48 @@
 import xmltodict
 
+from openml.study import OpenMLStudy
 from .._api_calls import _perform_api_call
 
+def _multitag_to_list(result_dict, tag):
+    if isinstance(result_dict[tag], list):
+        return result_dict[tag]
+    elif isinstance(result_dict[tag], dict):
+        return [result_dict[tag]]
+    else:
+        raise TypeError()
+
+
 def get_study(study_id):
-    xml_string = _perform_api_call("study/" % (study_id))
-    result_dict = xmltodict.parse(xml_string)
-    
-    pass
+    xml_string = _perform_api_call("study/%d" %(study_id))
+    result_dict = xmltodict.parse(xml_string)['oml:study']
+    id = int(result_dict['oml:id'])
+    name = result_dict['oml:name']
+    description = result_dict['oml:description']
+    creation_date = result_dict['oml:creation_date']
+    creator = result_dict['oml:creator']
+    tags = []
+    for tag in _multitag_to_list(result_dict, 'oml:tag'):
+        tags.append({'name': tag['oml:name'],
+                     'window_start': tag['oml:window_start'],
+                     'write_access': tag['oml:write_access']})
+
+    datasets = None
+    tasks = None
+    flows = None
+    setups = None
+
+    if 'oml:data' in result_dict:
+        datasets = [int(x) for x in result_dict['oml:data']['oml:data_id']]
+
+    if 'oml:tasks' in result_dict:
+        tasks = [int(x) for x in result_dict['oml:tasks']['oml:task_id']]
+
+    if 'oml:flows' in result_dict:
+        flows = [int(x) for x in result_dict['oml:flows']['oml:flow_id']]
+
+    if 'oml:setups' in result_dict:
+        setups = [int(x) for x in result_dict['oml:setups']['oml:setup_id']]
+
+    study = OpenMLStudy(id, name, description, creation_date, creator, tags,
+                        datasets, tasks, flows, setups)
+    return study
@@ -1,4 +1,3 @@
-import unittest
 import openml
 import openml.evaluations
 from openml.testing import TestBase
 
@@ -0,0 +1,16 @@
+import openml
+import openml.study
+from openml.testing import TestBase
+
+class TestStudyFunctions(TestBase):
+
+    def test_get_study(self):
+        openml.config.server = self.production_server
+
+        study_id = 34
+
+        study = openml.study.get_study(study_id)
+        self.assertEquals(len(study.data), 105)
+        self.assertEquals(len(study.tasks), 105)
+        self.assertEquals(len(study.flows), 27)
+        self.assertEquals(len(study.setups), 30)
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`		`-import unittest`
`2`	`1`	`import openml`
`3`	`2`	`import openml.evaluations`
`4`	`3`	`from openml.testing import TestBase`