e407b78506
* Remove six for PY2 compatability since our dependencies are not, PY2 is legacy. * Removed usages of utils from future/past libraries since they are deprecated. This includes 'from __future__ ...' and 'from past...' statements. * Removed compatibility imports for PY2, switched from six imports to standard library imports. * Removed utils code blocks that handle operations depending on PY2/3 versions. * Removed 'from __future__ ...' lines from templates/code generators * Used PY3 syntaxes in place of PY2 compatible blocks. eg: metaclass
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
# Copyright (c) 2019, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
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)"""
|
|
|
|
# Describe the columns
|
|
columns = {
|
|
0: {"label": "Column A", "fieldname": "column_a"},
|
|
1: {"label": "Column B", "fieldname": "column_b"},
|
|
2: {"label": "Column C", "fieldname": "column_c"}
|
|
}
|
|
|
|
# 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(columns, 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)
|