QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-18 15:44:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: upgrade black (#19410)

Browse Source
This commit is contained in:
Ville Brofeldt
2022-03-29 20:03:09 +03:00
committed by GitHub
parent 816a2c3e1e
commit a619cb4ea9
204 changed files with 2125 additions and 608 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/unit_tests/pandas_postprocessing/test_cum.py
View File

@@ -31,33 +31,49 @@ from tests.unit_tests.pandas_postprocessing.utils import series_to_list
def test_cum_should_not_side_effect():
_timeseries_df = timeseries_df.copy()
pp.cum(
df=timeseries_df, columns={"y": "y2"}, operator="sum",
df=timeseries_df,
columns={"y": "y2"},
operator="sum",
)
assert _timeseries_df.equals(timeseries_df)
def test_cum():
# create new column (cumsum)
post_df = pp.cum(df=timeseries_df, columns={"y": "y2"}, operator="sum",)
post_df = pp.cum(
df=timeseries_df,
columns={"y": "y2"},
operator="sum",
)
assert post_df.columns.tolist() == ["label", "y", "y2"]
assert series_to_list(post_df["label"]) == ["x", "y", "z", "q"]
assert series_to_list(post_df["y"]) == [1.0, 2.0, 3.0, 4.0]
assert series_to_list(post_df["y2"]) == [1.0, 3.0, 6.0, 10.0]
# overwrite column (cumprod)
post_df = pp.cum(df=timeseries_df, columns={"y": "y"}, operator="prod",)
post_df = pp.cum(
df=timeseries_df,
columns={"y": "y"},
operator="prod",
)
assert post_df.columns.tolist() == ["label", "y"]
assert series_to_list(post_df["y"]) == [1.0, 2.0, 6.0, 24.0]
# overwrite column (cummin)
post_df = pp.cum(df=timeseries_df, columns={"y": "y"}, operator="min",)
post_df = pp.cum(
df=timeseries_df,
columns={"y": "y"},
operator="min",
)
assert post_df.columns.tolist() == ["label", "y"]
assert series_to_list(post_df["y"]) == [1.0, 1.0, 1.0, 1.0]
# invalid operator
with pytest.raises(InvalidPostProcessingError):
pp.cum(
df=timeseries_df, columns={"y": "y"}, operator="abc",
df=timeseries_df,
columns={"y": "y"},
operator="abc",
)