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)
