Merge pull request #119 from openml/maint/improve_coverage

MAINT improve coverage: openml.datasets.functions

Author: amueller

openml/_api_calls.py

@@ -7,7 +7,7 @@
 from .exceptions import OpenMLServerError
 
 
-def _perform_api_call(call, data=None, file_dictionary=None,
+def _perform_api_call(call, file_dictionary=None,
                       file_elements=None, add_authentication=True):
     """
     Perform an API call at the OpenML server.
@@ -18,11 +18,14 @@ def _read_url(self, url, add_authentication=False, data=None, filePath=None):
     ----------
     call : str
         The API call. For example data/list
-    data : dict (default=None)
-        Dictionary containing data which will be sent to the OpenML
-        server via a POST request.
-    **kwargs
-        Further arguments which are appended as GET arguments.
+    file_dictionary : dict
+        Mapping of {filename: path} of files which should be uploaded to the
+        server.
+    file_elements : dict
+        Mapping of {filename: str} of strings which should be uploaded as
+        files to the server.
+    add_authentication : bool
+        Whether to add authentication (api key) to the request.
 
     Returns
     -------
@@ -36,16 +39,16 @@ def _read_url(self, url, add_authentication=False, data=None, filePath=None):
         url += "/"
     url += call
     if file_dictionary is not None or file_elements is not None:
-        return _read_url_files(url, data=data, file_dictionary=file_dictionary,
+        return _read_url_files(url, file_dictionary=file_dictionary,
                                file_elements=file_elements)
-    return _read_url(url, data=data)
+    return _read_url(url)
 
 
-def _read_url_files(url, data=None, file_dictionary=None, file_elements=None):
+def _read_url_files(url, file_dictionary=None, file_elements=None):
     """do a post request to url with data None, file content of
     file_dictionary and sending file_elements as files"""
-    if data is None:
-        data = {}
+
+    data = {}
     data['api_key'] = config.apikey
     if file_elements is None:
         file_elements = {}
@@ -78,9 +81,9 @@ def _read_url_files(url, data=None, file_dictionary=None, file_elements=None):
     return response.status_code, response.text
 
 
-def _read_url(url, data=None):
-    if data is None:
-        data = {}
+def _read_url(url):
+
+    data = {}
     data['api_key'] = config.apikey
 
     # Using requests.post sets header 'Accept-encoding' automatically to

openml/config.py

@@ -51,7 +51,7 @@ def _setup():
     set_cache_directory(cache_dir, private_dir)
 
 
-def set_cache_directory(cachedir, privatedir):
+def set_cache_directory(cachedir, privatedir=None):
     """Set module-wide cache directory.
 
     Sets the cache directory into which to download datasets, tasks etc.

openml/datasets/dataset.py

@@ -1,7 +1,8 @@
 import gzip
+import logging
 import os
 import sys
-import logging
+
 import arff
 
 import numpy as np
@@ -267,12 +268,17 @@ def publish(self):
         return_value : string
             xml return from server
         """
-        data = {'description': self._to_xml()}
+
+        file_elements = {'description': self._to_xml()}
+        file_dictionary = {}
+
         if self.data_file is not None:
-            return_code, return_value = _perform_api_call(
-                "/data/", data=data, file_dictionary={'dataset': self.data_file})
-        else:
-            return_code, return_value = _perform_api_call("/data/", data=data)
+            file_dictionary['dataset'] = self.data_file
+
+        return_code, return_value = _perform_api_call(
+            "/data/", file_dictionary=file_dictionary,
+            file_elements=file_elements)
+
         return return_code, return_value
 
     def _to_xml(self):