def write_csv_for_plot(results, outfile, metadata=None, column_name=None): ''' Writes a csv file to the specified outfile file-like object. Tweak the data to fill in gaps ''' csv_writer = csv.writer(outfile) if metadata != None: csv_writer.writerow(metadata) result_keys = results.keys() if column_name != None: result_keys.append(column_name) csv_writer.writerow(results.keys()) # Iterate, because csv.writerows pulls up all rows to a list if column_name == None: for r in gap_fill_by_3(results): csv_writer.writerow(r) else: extra = (None, ) for r in gap_fill_by_3(results): # have to convert to tuple because RowProxy does not support + csv_writer.writerow(tuple(r) + extra)
def write_csv_for_plot(results, outfile, metadata=None, column_name=None): ''' Writes a csv file to the specified outfile file-like object. Tweak the data to fill in gaps ''' csv_writer = csv.writer(outfile) if metadata != None: csv_writer.writerow(metadata) result_keys = results.keys() if column_name != None: result_keys.append(column_name) csv_writer.writerow(results.keys()) # Iterate, because csv.writerows pulls up all rows to a list if column_name == None: for r in gap_fill_by_3(results): csv_writer.writerow(r) else: extra = (None,) for r in gap_fill_by_3(results): # have to convert to tuple because RowProxy does not support + csv_writer.writerow(tuple(r) + extra)
def test_base_case_by_3(self): ssz = [('date', 100, None, None, None, None, None), ('date', 101, None, None, 0, None, None), ('date', None, 102, None, 0, None, None), ('date', 103, None, None, None, None, None), ('date', None, None, 104, None, None, 1), ('date', None, None, 105, None, None, 2), ('date', None, None, 106, None, None, None), ] gfg = gap_fill_by_3(ssz) l = list(gfg) expected = [ ('date', 100, None, None, None, None, None), ('date', 101, None, None, 0, None, None), ('date', 102, 102, None, 0, None, None), ('date', 103, 103, None, None, None, None), ('date', 104, None, 104, None, None, 1), ('date', None, None, 105, None, None, 2), ('date', None, None, 106, None, None, None), ] self.assertEquals(l, expected)
def test_base_case_by_3(self): ssz = [ ('date', 100, None, None, None, None, None), ('date', 101, None, None, 0, None, None), ('date', None, 102, None, 0, None, None), ('date', 103, None, None, None, None, None), ('date', None, None, 104, None, None, 1), ('date', None, None, 105, None, None, 2), ('date', None, None, 106, None, None, None), ] gfg = gap_fill_by_3(ssz) l = list(gfg) expected = [ ('date', 100, None, None, None, None, None), ('date', 101, None, None, 0, None, None), ('date', 102, 102, None, 0, None, None), ('date', 103, 103, None, None, None, None), ('date', 104, None, 104, None, None, 1), ('date', None, None, 105, None, None, 2), ('date', None, None, 106, None, None, None), ] self.assertEquals(l, expected)