From 91a275ef88ef29b5a74889682290c4c5401aeb67 Mon Sep 17 00:00:00 2001 From: iKintosh Date: Tue, 31 May 2022 10:35:42 +0300 Subject: [PATCH] fix style --- tests/test_analysis/test_plotters.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/tests/test_analysis/test_plotters.py b/tests/test_analysis/test_plotters.py index 4cf99b2ed..dab7b6ecb 100644 --- a/tests/test_analysis/test_plotters.py +++ b/tests/test_analysis/test_plotters.py @@ -265,7 +265,15 @@ def test_create_holidays_df_as_is(simple_df): def test_create_holidays_df_non_day_freq(): classic_df = generate_ar_df(periods=30, start_time="2020-01-01", n_segments=1, freq="H") ts = TSDataset.to_dataset(classic_df) - holidays = pd.DataFrame({"holiday": "Christmas", "ds": pd.to_datetime(["2020-01-01"], ), "upper_window": 3}) + holidays = pd.DataFrame( + { + "holiday": "Christmas", + "ds": pd.to_datetime( + ["2020-01-01"], + ), + "upper_window": 3, + } + ) df = _create_holidays_df(holidays, ts.index, as_is=False) assert df.sum().sum() == 4 @@ -288,7 +296,9 @@ def test_create_holidays_df_several_holidays(simple_df): def test_create_holidays_df_zero_windows(simple_df): - holidays = pd.DataFrame({"holiday": "Christmas", "ds": pd.to_datetime(["2020-01-07"]), "lower_window": 0, "upper_window": 0}) + holidays = pd.DataFrame( + {"holiday": "Christmas", "ds": pd.to_datetime(["2020-01-07"]), "lower_window": 0, "upper_window": 0} + ) df = _create_holidays_df(holidays, simple_df.index, as_is=False) assert df.sum().sum() == 1 assert df.loc["2020-01-07"].sum() == 1 @@ -304,4 +314,3 @@ def test_create_holidays_df_lower_window_positive(simple_df): holidays = pd.DataFrame({"holiday": "Christmas", "ds": pd.to_datetime(["2020-01-07"]), "lower_window": 1}) with pytest.raises(ValueError): _create_holidays_df(holidays, simple_df.index, as_is=False) -