41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
# Copyright (c) 2019, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
|
|
import unittest
|
|
|
|
import frappe
|
|
from frappe.desk.query_report import build_xlsx_data
|
|
import frappe.utils
|
|
|
|
|
|
class TestQueryReport(unittest.TestCase):
|
|
def test_xlsx_data_with_multiple_datatypes(self):
|
|
"""Test exporting report using rows with multiple datatypes (list, dict)"""
|
|
|
|
# Create mock data
|
|
data = frappe._dict()
|
|
data.columns = [
|
|
{"label": "Column A", "fieldname": "column_a"},
|
|
{"label": "Column B", "fieldname": "column_b", "width": 150},
|
|
{"label": "Column C", "fieldname": "column_c", "width": 100}
|
|
]
|
|
data.result = [
|
|
[1.0, 3.0, 5.5],
|
|
{"column_a": 22.1, "column_b": 21.8, "column_c": 30.2},
|
|
{"column_b": 5.1, "column_c": 9.5, "column_a": 11.1},
|
|
[3.0, 1.5, 7.5],
|
|
]
|
|
|
|
# Define the visible rows
|
|
visible_idx = [0, 2, 3]
|
|
|
|
# Build the result
|
|
xlsx_data, column_widths = build_xlsx_data(data, visible_idx, include_indentation=0)
|
|
|
|
self.assertEqual(type(xlsx_data), list)
|
|
self.assertEqual(len(xlsx_data), 4) # columns + data
|
|
# column widths are divided by 10 to match the scale that is supported by openpyxl
|
|
self.assertListEqual(column_widths, [0, 15, 10])
|
|
|
|
for row in xlsx_data:
|
|
self.assertEqual(type(row), list)
|