QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-19 16:14:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(viz): update nesting logic to handle multiple dimensions in PartitionViz (#32290)

Browse Source
This commit is contained in:
Damian Pendrak
2025-02-28 16:57:11 +01:00
committed by GitHub
parent 128c45e2d3
commit 6317a91541
2 changed files with 29 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tests/integration_tests/viz_tests.py
View File

@@ -442,6 +442,27 @@ class TestPartitionViz(SupersetTestCase):
assert 1 == len(nest[0]["children"][0]["children"])
assert 1 == len(nest[0]["children"][0]["children"][0]["children"])
def test_nest_values_returns_hierarchy_when_more_dimensions(self):
raw = {}
raw["category"] = ["a", "a", "a"]
raw["subcategory"] = ["a.2", "a.1", "a.2"]
raw["sub_subcategory"] = ["a.2.1", "a.1.1", "a.2.2"]
raw["metric1"] = [5, 10, 15]
raw["metric2"] = [50, 100, 150]
raw["metric3"] = [500, 1000, 1500]
df = pd.DataFrame(raw)
test_viz = viz.PartitionViz(Mock(), {})
groups = ["category", "subcategory", "sub_subcategory"]
levels = test_viz.levels_for("agg_sum", groups, df)
nest = test_viz.nest_values(levels)
assert 3 == len(nest)
for i in range(0, 3):
assert "metric" + str(i + 1) == nest[i]["name"]
assert 1 == len(nest[0]["children"])
assert 2 == len(nest[0]["children"][0]["children"])
assert 1 == len(nest[0]["children"][0]["children"][0]["children"])
assert 2 == len(nest[0]["children"][0]["children"][1]["children"])
def test_nest_procs_returns_hierarchy(self):
raw = {}
raw[DTTM_ALIAS] = [100, 200, 300, 100, 200, 300, 100, 200, 300]