def test_scatter_csv_export(): exporter = ExportCSV(MagicMock(), MagicMock()) out = StringIO() series = [ [ { 'data': [ { 'name': 'AT', 'x': 1, 'y': 1 } ], 'name': 'Austria' }, ] ] exporter.datapoints_n(out, series) out.seek(0) csv_output = out.read().split('\r\n') assert csv_output[0].split(',') == ['series', 'name', 'x', 'y']; assert csv_output[1].split(',') == ['Austria', 'AT', '1', '1']
def test_formatter_decision(): exporter = ExportCSV(MagicMock(), MagicMock()) series = [ { 'data': [ { 'code': '2000', 'name': 'year:2000', 'y': 1 } ], 'name': 'series1', 'notation': None }, ] exporter.datapoints = Mock() exporter.datapoints_n = Mock() exporter.request.form = { 'metadata': json.dumps({}), 'chart_data': json.dumps(series), 'chart_type': 'columns', 'annotations': json.dumps({}) } exporter.export() assert(exporter.datapoints.call_count == 1) exporter.request.form = { 'metadata': json.dumps({}), 'chart_data': json.dumps(series), 'chart_type': 'bubbles', 'annotations': json.dumps({}) } exporter.export() assert(exporter.datapoints_n.call_count == 1) exporter.request.form = { 'metadata': json.dumps({}), 'chart_data': json.dumps(series), 'chart_type': 'scatter', 'annotations': json.dumps({}) } exporter.export() assert(exporter.datapoints_n.call_count == 2)