QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-06-03 14:49:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: generate label map on the backend (#21124)

Browse Source
This commit is contained in:
Yongjie Zhao
2022-08-22 21:00:02 +08:00
committed by GitHub
parent 756ed0e36a
commit 11bf7b9125
8 changed files with 154 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/unit_tests/pandas_postprocessing/test_flatten.py
View File

@@ -156,3 +156,22 @@ def test_flat_integer_column_name():
}
)
)
def test_escape_column_name():
index = pd.to_datetime(["2021-01-01", "2021-01-02", "2021-01-03"])
index.name = "__timestamp"
columns = pd.MultiIndex.from_arrays(
[
["level1,value1", "level1,value2", "level1,value3"],
["level2, value1", "level2, value2", "level2, value3"],
],
names=["level1", "level2"],
)
df = pd.DataFrame(index=index, columns=columns, data=1)
assert list(pp.flatten(df).columns.values) == [
"__timestamp",
"level1\\,value1" + FLAT_COLUMN_SEPARATOR + "level2\\, value1",
"level1\\,value2" + FLAT_COLUMN_SEPARATOR + "level2\\, value2",
"level1\\,value3" + FLAT_COLUMN_SEPARATOR + "level2\\, value3",
]