class ExportImageTest(unittest.TestCase):
def setUp(self):
self.comm = Comm()
self.params = {'tree_id': '1234'}
@patch('itolapi.comm.requests')
def test_export_image(self, mock_requests):
mock_requests.post().content = 'asdf'
output = self.comm.export_image(self.params)
self.assertEqual(output, 'asdf')
class ItolExport:
EXPORT_FORMATS = ['png', 'svg', 'eps', 'ps', 'pdf', 'nexus', 'newick']
"""
Instantiate the itolexport class with empty params and empty server
"""
def __init__(self):
"""
Instantiate class
"""
self.params = dict({})
self.comm = Comm()
# Setting Export Parameters
def add_export_param_dict(self, param_dict):
"""
Add a dictionary of parameters to the parameters to be used when
exporting
@param: dictionary of parameters to be used
"""
self.params.update(param_dict)
def set_export_param_value(self, key, value):
"""
Add a value to the dictionary of parameters to be used when exporting
@param: dictionary of parameters to be used
"""
self.params[key] = value
def get_export_params(self):
"""
Get the dictionary of parameters to tbe used when exporting
@return: export the Parameters
"""
return self.params
# Do Exporting
def export(self, export_location):
"""
Call the export process
Calling this directly assumes that the export filetype is already set
in the export params
@param filelocation: the location to write the export to
@return: whether the export works
"""
output = self.comm.export_image(self.params)
file_handle = open(export_location, 'wb')
file_handle.write(output)
file_handle.close()
class ItolExport:
EXPORT_FORMATS = ['png', 'svg', 'eps', 'ps', 'pdf', 'nexus', 'newick']
"""
Instantiate the itolexport class with empty params and empty server
"""
def __init__(self):
"""
Instantiate class
"""
self.params = dict({})
self.comm = Comm()
# Setting Export Parameters
def add_export_param_dict(self, param_dict):
"""
Add a dictionary of parameters to the parameters to be used when
exporting
@param: dictionary of parameters to be used
"""
self.params.update(param_dict)
def set_export_param_value(self, key, value):
"""
Add a value to the dictionary of parameters to be used when exporting
@param: dictionary of parameters to be used
"""
self.params[key] = value
def get_export_params(self):
"""
Get the dictionary of parameters to tbe used when exporting
@return: export the Parameters
"""
return self.params
# Do Exporting
def export(self, export_location):
"""
Call the export process
Calling this directly assumes that the export filetype is already set
in the export params
@param filelocation: the location to write the export to
@return: whether the export works
"""
output = self.comm.export_image(self.params)
file_handle = open(export_location, 'wb')
file_handle.write(output)
file_handle.close()
