def clean_excel(self): excel = self.cleaned_data['excel'] try: parse_xlsx(excel) except RuntimeError: raise ValidationError('excel file is malformed') except BadZipfile: raise ValidationError('excel file is not an excel(xlsx) file') return excel
def test_parse_xlsx(self): plot_type, labels, data = parse_xlsx(os.path.join(TEST_DATA, 'columns.xlsx')) assert plot_type == COLUMNS assert data[YVALUES] == [1.0, 3.0, 4.0, 5.0, 5.0] assert labels[XLABEL] == 'xlabel2' plot_type, labels, data = parse_xlsx(os.path.join(TEST_DATA, 'scatter.xlsx')) plot_type = SCATTER assert data[YVALUES] == [1.0, 3.0, 4.0, 5.0, 5.0] assert labels[XLABEL] == 'xlabel1'
def combined_experiment_excel_data(experiments_by_type): for type_name, exps in experiments_by_type.items(): excel_data = {} for exp in exps: for excel_prop in exp.excel_props: exp_name = excel_prop.experiment.uniquename plot_type, labels, data = parse_xlsx(excel_prop.excel.path) if plot_type != COLUMNS: continue excel_data[exp_name] = labels, data yield type_name, excel_data
def test_empty_file_excel(self): _, labels, data = parse_xlsx(os.path.join(TEST_DATA, 'empty_file.xlsx'))
def test_draw_scatter(self): _, labels, data = parse_xlsx(os.path.join(TEST_DATA, 'scatter.xlsx')) out_fhand = NamedTemporaryFile(suffix='.svg') draw_scatter(labels, data, out_fhand) raw_input(out_fhand.name)