From e2dc1cc145430a1de0022664015c4a9cde73ce00 Mon Sep 17 00:00:00 2001 From: FBruzzesi Date: Tue, 2 Jul 2024 09:05:35 +0200 Subject: [PATCH] lint --- tests/test_pandas_utils/test_pandas_utils.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tests/test_pandas_utils/test_pandas_utils.py b/tests/test_pandas_utils/test_pandas_utils.py index 841a2e12..fd501574 100644 --- a/tests/test_pandas_utils/test_pandas_utils.py +++ b/tests/test_pandas_utils/test_pandas_utils.py @@ -38,11 +38,10 @@ def test_add_lags_wrong_inputs(data, frame_func): add_lags(invalid_df, ["X1"], 1) -@pytest.mark.parametrize("frame_func", [ - pd.DataFrame, - lambda data: pl.DataFrame(data, strict=False), - lambda data: pl.LazyFrame(data, strict=False) -]) +@pytest.mark.parametrize( + "frame_func", + [pd.DataFrame, lambda data: pl.DataFrame(data, strict=False), lambda data: pl.LazyFrame(data, strict=False)], +) def test_add_lags_correct_df(data, frame_func): test_df = frame_func(data) expected = frame_func({"X1": [1, 2], "X2": ["178", "154"], "X1-1": [0, 1]})