def test_dicts_to_csv(self): output_file = StringIO.StringIO() obj1 = {"a": "v1", "b": "v2"} obj2 = {"a": "v\"'3", "b": "v4"} obj3 = {"a": "v5", "c": "v6"} dicts_to_csv([obj1, obj2, obj3], output_file) result = output_file.getvalue().strip("\r\n") self.assertEquals('a,b,c\r\nv1,v2,\r\n"v""\'3",v4,\r\nv5,,v6', result)
def test_dicts_to_csv(self): output_file = StringIO.StringIO() obj1 = { 'a': 'v1', 'b': 'v2'} obj2 = { 'a': 'v"\'3', 'b': 'v4'} obj3 = { 'a': 'v5', 'c': 'v6'} dicts_to_csv([obj1, obj2, obj3], output_file) result = output_file.getvalue().strip('\r\n') self.assertEquals('a,b,c\r\nv1,v2,\r\n"v""\'3",v4,\r\nv5,,v6', result)
def mergeResult(resultBase, outputFile): list_dir = os.walk(resultBase) output = open(outputFile, "w") result_list = [] for root, dirs, files in list_dir: for f in files: if f.startswith("."): continue file_open = open(root + "/" + f, "rb") setting = pickle.load(file_open) eval_list = setting.evalMethods eval_num = len(eval_list) result_num = len(eval_list[0].value) for i in range(result_num): abstract = {} abstract.update(setting.abstract) for eval_index in range(eval_num): abstract.update({eval_list[eval_index].name: eval_list[eval_index].value[i]}) result_list.append(abstract) dicts_to_csv(result_list, output)